diff --git a/image_classifier/image_classifier.py b/image_classifier/image_classifier.py index 50a8afc..a1884a6 100755 --- a/image_classifier/image_classifier.py +++ b/image_classifier/image_classifier.py @@ -86,7 +86,6 @@ 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 @@ -98,8 +97,9 @@ 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: - names.append(known_person['name']) + for encoding in encodings: + if known_person['encoding'] == encoding: + names.append(known_person['name']) except PIL.UnidentifiedImageError as error: self._log.debug(f"File '{file}' don't seem to be an image.") return False