wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-13 18:47:29 +08:00
Code Issues Actions 22 Packages Projects Releases Wiki Activity

Make mask optional

Browse Source
This commit is contained in:
missionfloyd 2023-04-13 21:19:10 -06:00
parent 71f939c3ed
commit 46d0fa6424
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
comfy_extras/nodes_post_processing.py
View File

@ -428,10 +428,12 @@ class Composite:
"required": { "required": {
"image_a": ("IMAGE",), "image_a": ("IMAGE",),
"image_b": ("IMAGE",), "image_b": ("IMAGE",),
"mask": ("MASK",),
"x": ("INT", {"default": 0, "min": -MAX_RESOLUTION, "max": MAX_RESOLUTION}), "x": ("INT", {"default": 0, "min": -MAX_RESOLUTION, "max": MAX_RESOLUTION}),
"y": ("INT", {"default": 0, "min": -MAX_RESOLUTION, "max": MAX_RESOLUTION}), "y": ("INT", {"default": 0, "min": -MAX_RESOLUTION, "max": MAX_RESOLUTION}),
}, },
"optional": {
"mask": ("MASK",),
}
} }
RETURN_TYPES = ("IMAGE",) RETURN_TYPES = ("IMAGE",)
@ -439,11 +441,14 @@ class Composite:
CATEGORY = "image/postprocessing" CATEGORY = "image/postprocessing"
def composite(self, image_a: torch.Tensor, image_b: torch.Tensor, mask: torch.Tensor, x: int, y: int): def composite(self, image_a: torch.Tensor, image_b: torch.Tensor, x: int, y: int, mask: torch.Tensor = None):
batch_size, height, width, _ = image_a.shape batch_size, height, width, _ = image_a.shape
result = torch.zeros_like(image_a) result = torch.zeros_like(image_a)
for b in range(batch_size): for b in range(batch_size):
if mask is None:
mask = torch.ones(image_b.shape[1:3])
img_a = (image_a[b] * 255).to(torch.uint8).numpy() img_a = (image_a[b] * 255).to(torch.uint8).numpy()
img_b = (image_b[b] * 255).to(torch.uint8).numpy() img_b = (image_b[b] * 255).to(torch.uint8).numpy()
img_mask = (mask * 255).to(torch.uint8).numpy() img_mask = (mask * 255).to(torch.uint8).numpy()