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

add debug

Browse source
This commit is contained in:
Antonio J. Delgado 2021-09-06 16:35:37 +03:00
parent c226fc6e90
commit 0b6bfa672c
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
image_classifier/image_classifier.py
View file

@ -86,6 +86,7 @@ class image_classifier:
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['encoding'] = face_recognition.face_encodings(person['filename'])[0]
print(type(person['encoding']))
known_people.append(person)
return known_people
@ -97,7 +98,7 @@ class image_classifier:
encodings = face_recognition.face_encodings(image)
self._log.debug(f"Found {len(encodings)} faces.")
for known_person in self.known_people:
if known_person['encoding'] in encodings.tolist():
if known_person['encoding'] in encodings:
names.append(known_person['name'])
except PIL.UnidentifiedImageError as error:
self._log.debug(f"File '{file}' don't seem to be an image.")