Compare commits

...

2 Commits

Author SHA1 Message Date
dzqdzq
43b7bf823f
Merge e8e5496b4d into e346d8584e 2025-04-10 00:23:07 +02:00
dzqdzq
e8e5496b4d feat(nodes.py): Add support for alpha channel output in LoadImage node 2025-02-17 01:50:14 +08:00

View File

@ -1655,14 +1655,17 @@ class LoadImage:
input_dir = folder_paths.get_input_directory()
files = [f for f in os.listdir(input_dir) if os.path.isfile(os.path.join(input_dir, f))]
return {"required":
{"image": (sorted(files), {"image_upload": True})},
{
"image": (sorted(files), {"image_upload": True}),
"out_alpha": ("BOOLEAN", {"default": False}),
}
}
CATEGORY = "image"
RETURN_TYPES = ("IMAGE", "MASK")
FUNCTION = "load_image"
def load_image(self, image):
def load_image(self, image, out_alpha):
image_path = folder_paths.get_annotated_filepath(image)
img = node_helpers.pillow(Image.open, image_path)
@ -1678,7 +1681,7 @@ class LoadImage:
if i.mode == 'I':
i = i.point(lambda i: i * (1 / 255))
image = i.convert("RGB")
image = i.convert("RGBA" if out_alpha else "RGB")
if len(output_images) == 0:
w = image.size[0]