From e8e5496b4d4db520215468a996c50b69f7f0abf9 Mon Sep 17 00:00:00 2001
From: dzqdzq <635587322@qq.com>
Date: Mon, 17 Feb 2025 01:50:14 +0800
Subject: [PATCH] feat(nodes.py): Add support for alpha channel output in
 LoadImage node

---
 nodes.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/nodes.py b/nodes.py
index 7defb60b7..bdd0ab962 100644
--- a/nodes.py
+++ b/nodes.py
@@ -1648,14 +1648,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)
@@ -1671,7 +1674,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]