adelgado/image_classifier
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Use basename

Browse source
This commit is contained in:
Antonio J. Delgado 2021-09-06 16:56:12 +03:00
parent e121afdbea
commit 45a4591c57
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
image_classifier/image_classifier.py
View file

@ -113,7 +113,7 @@ class image_classifier:
for known_person in self.known_people: for known_person in self.known_people:
for encoding in encodings: for encoding in encodings:
if face_recognition.compare_faces([known_person['encoding']], encoding)[0]: 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: except PIL.UnidentifiedImageError as error:
self._log.debug(f"File '{file}' don't seem to be an image.") self._log.debug(f"File '{file}' don't seem to be an image.")
return False return False