|
|
|
@ -10,7 +10,7 @@ import time
|
|
|
|
|
import random |
|
|
|
|
import logging |
|
|
|
|
|
|
|
|
|
from PIL import Image, ImageOps, ImageSequence |
|
|
|
|
from PIL import Image, ImageOps, ImageSequence, ImageFile |
|
|
|
|
from PIL.PngImagePlugin import PngInfo |
|
|
|
|
|
|
|
|
|
import numpy as np |
|
|
|
@ -1462,7 +1462,17 @@ class LoadImage:
|
|
|
|
|
output_images = [] |
|
|
|
|
output_masks = [] |
|
|
|
|
for i in ImageSequence.Iterator(img): |
|
|
|
|
i = ImageOps.exif_transpose(i) |
|
|
|
|
prev_value = None |
|
|
|
|
try: |
|
|
|
|
i = ImageOps.exif_transpose(i) |
|
|
|
|
except OSError: |
|
|
|
|
prev_value = ImageFile.LOAD_TRUNCATED_IMAGES |
|
|
|
|
ImageFile.LOAD_TRUNCATED_IMAGES = True |
|
|
|
|
i = ImageOps.exif_transpose(i) |
|
|
|
|
finally: |
|
|
|
|
if prev_value is not None: |
|
|
|
|
ImageFile.LOAD_TRUNCATED_IMAGES = prev_value |
|
|
|
|
|
|
|
|
|
if i.mode == 'I': |
|
|
|
|
i = i.point(lambda i: i * (1 / 255)) |
|
|
|
|
image = i.convert("RGB") |
|
|
|
|