wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-08 21:30:50 +08:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

Merge e8e5496b4d into d9a76cf66e

Browse Source
This commit is contained in:
浪浪辛 2025-12-26 15:47:27 +08:00 committed by GitHub
commit 64b4c776e1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
nodes.py
View File

@ -1647,14 +1647,17 @@ class LoadImage:
files = [f for f in os.listdir(input_dir) if os.path.isfile(os.path.join(input_dir, f))] files = [f for f in os.listdir(input_dir) if os.path.isfile(os.path.join(input_dir, f))]
files = folder_paths.filter_files_content_types(files, ["image"]) files = folder_paths.filter_files_content_types(files, ["image"])
return {"required": return {"required":
{"image": (sorted(files), {"image_upload": True})}, {
"image": (sorted(files), {"image_upload": True}),
"out_alpha": ("BOOLEAN", {"default": False}),
}
} }
CATEGORY = "image" CATEGORY = "image"
RETURN_TYPES = ("IMAGE", "MASK") RETURN_TYPES = ("IMAGE", "MASK")
FUNCTION = "load_image" FUNCTION = "load_image"
def load_image(self, image): def load_image(self, image, out_alpha):
image_path = folder_paths.get_annotated_filepath(image) image_path = folder_paths.get_annotated_filepath(image)
img = node_helpers.pillow(Image.open, image_path) img = node_helpers.pillow(Image.open, image_path)
@ -1670,7 +1673,7 @@ class LoadImage:
if i.mode == 'I': if i.mode == 'I':
i = i.point(lambda i: i * (1 / 255)) 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: if len(output_images) == 0:
w = image.size[0] w = image.size[0]