Ignore all frames except the first one for MPO format.

This commit is contained in:
bigcat88 2025-12-31 08:35:56 +02:00
parent 6ca3d5c011
commit a7acc6533c
No known key found for this signature in database
GPG Key ID: 1F0BF0EC3CF22721

View File

@ -1663,8 +1663,6 @@ class LoadImage:
output_masks = [] output_masks = []
w, h = None, None w, h = None, None
excluded_formats = ['MPO']
for i in ImageSequence.Iterator(img): for i in ImageSequence.Iterator(img):
i = node_helpers.pillow(ImageOps.exif_transpose, i) i = node_helpers.pillow(ImageOps.exif_transpose, i)
@ -1692,7 +1690,10 @@ class LoadImage:
output_images.append(image) output_images.append(image)
output_masks.append(mask.unsqueeze(0)) output_masks.append(mask.unsqueeze(0))
if len(output_images) > 1 and img.format not in excluded_formats: if img.format == "MPO":
break # ignore all frames except the first one for MPO format
if len(output_images) > 1:
output_image = torch.cat(output_images, dim=0) output_image = torch.cat(output_images, dim=0)
output_mask = torch.cat(output_masks, dim=0) output_mask = torch.cat(output_masks, dim=0)
else: else: