diff --git a/image_classifier/image_classifier.py b/image_classifier/image_classifier.py index c2276a0..1a82280 100755 --- a/image_classifier/image_classifier.py +++ b/image_classifier/image_classifier.py @@ -79,8 +79,8 @@ class image_classifier: for entry in faces_items: if not entry.name.startswith('.') and entry.is_file(): person = dict() - person['filename'] = face_recognition.load_image_file(entry.name) - person['name'] = os.path.splitext(entry.name)[0] + 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] known_people.append(person) return known_people