Commit 14e55a33 authored by w-e-w's avatar w-e-w
Browse files

print PIL.UnidentifiedImageError

parent 335428c2
Loading
Loading
Loading
Loading
+2 −1
Original line number Diff line number Diff line
@@ -48,7 +48,8 @@ def process_batch(p, input_dir, output_dir, inpaint_mask_dir, args):

        try:
            img = Image.open(image)
        except UnidentifiedImageError:
        except UnidentifiedImageError as e:
            print(e)
            continue
        # Use the EXIF orientation of photos taken by smartphones.
        img = ImageOps.exif_transpose(img)