diff --git a/image_classifier/image_classifier.py b/image_classifier/image_classifier.py index cd0687a..c4790b7 100755 --- a/image_classifier/image_classifier.py +++ b/image_classifier/image_classifier.py @@ -17,6 +17,7 @@ from logging.handlers import SysLogHandler import face_recognition import pyexiv2 import PIL +import cv2 class CustomFormatter(logging.Formatter): """Logging colored formatter, adapted from https://stackoverflow.com/a/56944256/3638629""" @@ -176,7 +177,7 @@ class image_classifier: def is_image(self, file): try: - im = PIL.Image.open(file) + image_object = PIL.Image.open(file) except PIL.UnidentifiedImageError as error: self._log.debug(f"File '{file}' is not an image recognizable by PIL. {error}") return False