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

Change to loop on list

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

6
image_classifier/image_classifier.py
View file

@ -86,7 +86,6 @@ class image_classifier:
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.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]
print(type(person['encoding']))
known_people.append(person) known_people.append(person)
return known_people return known_people
@ -98,8 +97,9 @@ class image_classifier:
encodings = face_recognition.face_encodings(image) encodings = face_recognition.face_encodings(image)
self._log.debug(f"Found {len(encodings)} faces.") self._log.debug(f"Found {len(encodings)} faces.")
for known_person in self.known_people: for known_person in self.known_people:
if known_person['encoding'] in encodings: for encoding in encodings:
names.append(known_person['name']) if known_person['encoding'] == encoding:
names.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