diff --git a/image_classifier/image_classifier.py b/image_classifier/image_classifier.py index 61fefff..4763d66 100755 --- a/image_classifier/image_classifier.py +++ b/image_classifier/image_classifier.py @@ -113,7 +113,7 @@ class image_classifier: for known_person in self.known_people: for encoding in encodings: if face_recognition.compare_faces([known_person['encoding']], encoding)[0]: - people.append(known_person['name']) + people.append(os.path.basename(known_person['name'])) except PIL.UnidentifiedImageError as error: self._log.debug(f"File '{file}' don't seem to be an image.") return False