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

Add check for no image

Browse source
This commit is contained in:
Antonio J. Delgado 2021-09-06 16:17:53 +03:00
parent 3a0cb54fa0
commit 660db24dec
1 changed files with 8 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
image_classifier/image_classifier.py
View file

@ -66,6 +66,7 @@ class image_classifier:
move it to the folder of the day''' move it to the folder of the day'''
self._log.debug(f"Processing file '{file}'...") self._log.debug(f"Processing file '{file}'...")
people = self.find_faces(file) people = self.find_faces(file)
if people:
with open(file, 'rb') as image_file: with open(file, 'rb') as image_file:
exif_info = exif.Image(image_file) exif_info = exif.Image(image_file)
if exif_info.has_exif: if exif_info.has_exif:
@ -99,6 +100,7 @@ class image_classifier:
names.append(known_person['name']) names.append(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 names return names
def _init_log(self): def _init_log(self):