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

Add catch for no image

Browse source
This commit is contained in:
Antonio J. Delgado 2021-09-06 16:14:22 +03:00
parent 2a668b32c5
commit 538f59af6d
1 changed files with 9 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
image_classifier/image_classifier.py
View file

@ -79,7 +79,7 @@ class image_classifier:
with os.scandir(self.faces_directory) as faces_items:
for entry in faces_items:
if not entry.name.startswith('.') and entry.is_file():
self._log.debug(f"Detecting people in known faces '{entry.name}'...")
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]
@ -89,12 +89,15 @@ class image_classifier:
def find_faces(self, file):
''' Find faces in an image/video file '''
image = face_recognition.load_image_file(file)
encodings = face_recognition.face_encodings(image)
names = list()
for known_person in self.known_people:
if known_person['encoding'] in encodings:
names.append(known_person['name'])
try:
image = face_recognition.load_image_file(file)
encodings = face_recognition.face_encodings(image)
for known_person in self.known_people:
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.")
return names
def _init_log(self):