298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-01-11 02:15:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make the LoadImage node output the alpha as a mask.

Browse Source
This commit is contained in:
comfyanonymous 2023-03-09 14:07:55 -05:00
parent b94ed02aae
commit 99eb777f5d
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
nodes.py
View File

@ -815,7 +815,7 @@ class LoadImage:
CATEGORY = "image" CATEGORY = "image"
RETURN_TYPES = ("IMAGE",) RETURN_TYPES = ("IMAGE", "MASK")
FUNCTION = "load_image" FUNCTION = "load_image"
def load_image(self, image): def load_image(self, image):
image_path = os.path.join(self.input_dir, image) image_path = os.path.join(self.input_dir, image)
@ -823,7 +823,12 @@ class LoadImage:
image = i.convert("RGB") image = i.convert("RGB")
image = np.array(image).astype(np.float32) / 255.0 image = np.array(image).astype(np.float32) / 255.0
image = torch.from_numpy(image)[None,] image = torch.from_numpy(image)[None,]
return (image,) if 'A' in i.getbands():
mask = np.array(i.getchannel('A')).astype(np.float32) / 255.0
mask = 1. - torch.from_numpy(mask)
else:
mask = torch.zeros((64,64), dtype=torch.float32, device="cpu")
return (image, mask)
@classmethod @classmethod
def IS_CHANGED(s, image): def IS_CHANGED(s, image):