wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-10 05:22:34 +08:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

Add ImageAlphaComposite

Browse Source
This commit is contained in:
Saquib Alam 2023-08-08 20:08:57 +05:30 committed by GitHub
parent d8fa58a532
commit e1349163af
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 19 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
nodes.py
View File

@ -1434,6 +1434,25 @@ class ImageScale:
s = s.movedim(1,-1) s = s.movedim(1,-1)
return (s,) return (s,)
class ImageAlphaComposite:
@classmethod
def INPUT_TYPES(s):
return {"required": { "image1": ("IMAGE",),
"mask1": ("MASK",),
"image2": ("IMAGE",),
"mask2": ("MASK", {"default": None}),
}
CATEGORY = "image"
RETURN_TYPES = ("IMAGE", "MASK")
FUNCTION = "alpha_composite"
def alpha_composite(self, image1, mask1, image2, mask2):
if not mask2:
mask2 = torch.zeros((64,64), dtype=torch.float32, device="cpu")
image = image1 * mask1 + image2 * (1.0 - mask1)
return (image, mask1)
class ImageScaleBy: class ImageScaleBy:
upscale_methods = ["nearest-exact", "bilinear", "area", "bicubic"] upscale_methods = ["nearest-exact", "bilinear", "area", "bicubic"]