wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-11 17:52:46 +08:00
Code Issues Actions 18 Packages Projects Releases Wiki Activity

Don't need to create an empty mask

Browse Source
This commit is contained in:
missionfloyd 2023-04-14 13:32:21 -06:00
parent 2bd435c25c
commit cf2b046ec3
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
comfy_extras/nodes_post_processing.py
View File

@ -452,17 +452,18 @@ class Composite:
result = torch.zeros_like(base_image) result = torch.zeros_like(base_image)
for b in range(batch_size): for b in range(batch_size):
if mask is None:
mask = torch.ones(overlay_image.shape[1:3])
img_a = (base_image[b] * 255).to(torch.uint8).numpy() img_a = (base_image[b] * 255).to(torch.uint8).numpy()
img_b = (overlay_image[b] * 255).to(torch.uint8).numpy() img_b = (overlay_image[b] * 255).to(torch.uint8).numpy()
img_mask = (mask * 255).to(torch.uint8).numpy()
pil_base_image = Image.fromarray(img_a, mode='RGB') pil_base_image = Image.fromarray(img_a, mode='RGB')
pil_overlay_image = Image.fromarray(img_b, mode='RGB') pil_overlay_image = Image.fromarray(img_b, mode='RGB')
pil_image_mask = Image.fromarray(img_mask, mode='L')
if pil_image_mask.size != pil_overlay_image.size: if mask is None:
pil_image_mask = pil_image_mask.resize(pil_overlay_image.size, resamplers[resample]) pil_image_mask = mask
else:
img_mask = (mask * 255).to(torch.uint8).numpy()
pil_image_mask = Image.fromarray(img_mask, mode='L')
if pil_image_mask.size != pil_overlay_image.size:
pil_image_mask = pil_image_mask.resize(pil_overlay_image.size, resamplers[resample])
pil_base_image.paste(pil_overlay_image, (x, y), pil_image_mask) pil_base_image.paste(pil_overlay_image, (x, y), pil_image_mask)