Commit 99eb777f authored by comfyanonymous's avatar comfyanonymous
Browse files

Make the LoadImage node output the alpha as a mask.

parent b94ed02a
...@@ -815,7 +815,7 @@ class LoadImage: ...@@ -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: ...@@ -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):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment