Browse Source

Adding default alpha when splitting RGB images

pull/1578/head
MoonRide303 1 year ago
parent
commit
585fb0475b
  1. 2
      comfy_extras/nodes_compositing.py

2
comfy_extras/nodes_compositing.py

@ -153,7 +153,7 @@ class SplitImageWithAlpha:
def split_image_with_alpha(self, image: torch.Tensor):
out_images = [i[:,:,:3] for i in image]
out_alphas = [i[:,:,3] for i in image]
out_alphas = [i[:,:,3] if i.shape[2] > 3 else torch.ones_like(i[:,:,0]) for i in image]
result = (torch.stack(out_images), torch.stack(out_alphas))
return result

Loading…
Cancel
Save