mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-01-07 21:00:49 +08:00
Ignore all frames except the first one for MPO format.
This commit is contained in:
parent
6ca3d5c011
commit
a7acc6533c
7
nodes.py
7
nodes.py
@ -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:
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user