From 0b6bfa672c9d40e0d20335f34f5687fd57a76e12 Mon Sep 17 00:00:00 2001 From: "Antonio J. Delgado" Date: Mon, 6 Sep 2021 16:35:37 +0300 Subject: [PATCH] add debug --- image_classifier/image_classifier.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/image_classifier/image_classifier.py b/image_classifier/image_classifier.py index 7bd5d1d..50a8afc 100755 --- a/image_classifier/image_classifier.py +++ b/image_classifier/image_classifier.py @@ -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.")