ImageRemoveAlphaNode

Author: yanhuifair

__init__.py

@@ -35,6 +35,7 @@
 from .nodes.image_utility import ImagesInfoNode
 from .nodes.image_utility import ModulationNode
 from .nodes.image_utility import ModulationDirectionNode
+from .nodes.image_utility import ImageRemoveAlphaNode
 
 
 from .nodes.utility import PrintAnyNode
@@ -109,6 +110,7 @@
     "ASCIICharNode": ASCIICharNode,
     "ModulationNode": ModulationNode,
     "ModulationDirectionNode": ModulationDirectionNode,
+    "ImageRemoveAlphaNode": ImageRemoveAlphaNode,
 }
 
 NODE_DISPLAY_NAME_MAPPINGS = {
@@ -163,6 +165,7 @@
     "ASCIICharNode": "ASCII Art Text",
     "ModulationNode": "Modulation",
     "ModulationDirectionNode": "Modulation Direction",
+    "ImageRemoveAlphaNode": "Image Remove Alpha",
 }
 
 

nodes/image_utility.py

@@ -71,6 +71,11 @@ def tensor_to_pil(tensor):
     return toPIL(tensor.permute(2, 0, 1))
 
 
+def tensor_to_pil_single(tensor):
+    # [H, W] to [ H, W]
+    return toPIL(tensor)
+
+
 def tensor_to_batch(tensor, h, w, c):
     tensor = torch.cat(tensor)
     tensor = tensor.reshape(-1, h, w, c)
@@ -982,3 +987,37 @@ def INPUT_TYPES(cls):
 
     def node_function(self, direction):
         return (direction,)
+
+
+class ImageRemoveAlphaNode:
+    def __init__(self):
+        pass
+
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "images": (IO.IMAGE, {"defaultInput": True}),
+                "masks": (IO.MASK, {"defaultInput": True}),
+                "fill_color": (IO.STRING, {"default": "#FFFFFF"}),
+            }
+        }
+
+    FUNCTION = "node_function"
+    CATEGORY = "Fair/image"
+    RETURN_TYPES = (IO.IMAGE,)
+    RETURN_NAMES = ("images",)
+
+    def node_function(self, images, masks, fill_color):
+        out_images = []
+        for image, mask in zip(images, masks):
+            pil = tensor_to_pil(image)
+            pil_mask = tensor_to_pil_single(mask)
+
+            new_pil = Image.new("RGBA", pil.size, fill_color)
+            new_pil.paste(pil, pil_mask)
+
+            image = pil_to_tensor(new_pil)
+            out_images.append(image)
+        out_images = torch.stack(out_images, dim=0)
+        return (out_images,)

pyproject.toml

@@ -63,7 +63,7 @@ Python Script
 
 Load LoRA Dual
 """
-version = "1.0.77"
+version = "1.0.78"
 license = { file = "LICENSE" }
 dependencies = [
     "googletrans",