check_dpi: catch Pillow choking on faulty img, return 230

pull/33/head
Konstantin Baierer 4 years ago
parent ff265eee5c
commit 6c8852eb04

@ -15,13 +15,13 @@ def pil2cv(img):
return cvtColor(pil_as_np_array, color_conversion) return cvtColor(pil_as_np_array, color_conversion)
def check_dpi(img): def check_dpi(img):
try:
if isinstance(img, Image.__class__): if isinstance(img, Image.__class__):
pil_image = img pil_image = img
elif isinstance(img, str): elif isinstance(img, str):
pil_image = Image.open(img) pil_image = Image.open(img)
else: else:
pil_image = cv2pil(img) pil_image = cv2pil(img)
try:
exif = OcrdExif(pil_image) exif = OcrdExif(pil_image)
resolution = exif.resolution resolution = exif.resolution
if resolution == 1: if resolution == 1:

Loading…
Cancel
Save