diff --git a/image_classifier/image_classifier.py b/image_classifier/image_classifier.py index 4763d66..412c6d0 100755 --- a/image_classifier/image_classifier.py +++ b/image_classifier/image_classifier.py @@ -98,7 +98,7 @@ class image_classifier: self._log.debug(f"Detecting known person in file '{entry.name}'...") person = dict() 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.basename(os.path.splitext(self.faces_directory + os.sep + entry.name)[0]) person['encoding'] = face_recognition.face_encodings(person['filename'])[0] known_people.append(person) return known_people @@ -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(os.path.basename(known_person['name'])) + people.append(known_person['name']) except PIL.UnidentifiedImageError as error: self._log.debug(f"File '{file}' don't seem to be an image.") return False