Ignore all frames except the first one for MPO format. (#11569)

This commit is contained in:
Alexander Piskun 2026-01-02 10:35:34 +02:00 committed by GitHub
parent 65cfcf5b1b
commit 9e5f677746
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1663,8 +1663,6 @@ class LoadImage:
output_masks = []
w, h = None, None
excluded_formats = ['MPO']
for i in ImageSequence.Iterator(img):
i = node_helpers.pillow(ImageOps.exif_transpose, i)
@ -1692,7 +1690,10 @@ class LoadImage:
output_images.append(image)
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_mask = torch.cat(output_masks, dim=0)
else: