adelgado/image_classifier
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

change to loopFix comparehange place to get basename

Browse source
This commit is contained in:
Antonio J. Delgado 2021-09-06 16:56:41 +03:00
parent 45a4591c57
commit 77d4b2e9c4
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
image_classifier/image_classifier.py
View file

@ -98,7 +98,7 @@ class image_classifier:
self._log.debug(f"Detecting known person in file '{entry.name}'...") self._log.debug(f"Detecting known person in file '{entry.name}'...")
person = dict() person = dict()
person['filename'] = face_recognition.load_image_file(self.faces_directory + os.sep + entry.name) person['filename'] = face_recognition.load_image_file(self.faces_directory + os.sep + entry.name)
person['name'] = os.path.splitext(self.faces_directory + os.sep + entry.name)[0] person['name'] = os.path.basename(os.path.splitext(self.faces_directory + os.sep + entry.name)[0])
person['encoding'] = face_recognition.face_encodings(person['filename'])[0] person['encoding'] = face_recognition.face_encodings(person['filename'])[0]
known_people.append(person) known_people.append(person)
return known_people return known_people
@ -113,7 +113,7 @@ class image_classifier:
for known_person in self.known_people: for known_person in self.known_people:
for encoding in encodings: for encoding in encodings:
if face_recognition.compare_faces([known_person['encoding']], encoding)[0]: if face_recognition.compare_faces([known_person['encoding']], encoding)[0]:
people.append(os.path.basename(known_person['name'])) people.append(known_person['name'])
except PIL.UnidentifiedImageError as error: except PIL.UnidentifiedImageError as error:
self._log.debug(f"File '{file}' don't seem to be an image.") self._log.debug(f"File '{file}' don't seem to be an image.")
return False return False