xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2026-01-28 08:07:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update nodes.py

Browse Source
This commit is contained in:
Kijai 2024-01-18 15:54:25 +02:00
parent e1391ddf9f
commit b9c12ff13f
1 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
nodes.py
View File

@ -1237,12 +1237,10 @@ class SaveImageWithAlpha:
a = 255. * alpha.cpu().numpy()
img = Image.fromarray(np.clip(i, 0, 255).astype(np.uint8))
if a.shape == img.size[::-1]: # Check if the mask has the same size as the image
print("Applying mask")
a = np.clip(a, 0, 255).astype(np.uint8)
img.putalpha(Image.fromarray(a, mode='L'))
else:
raise ValueError("SaveImageWithAlpha: Mask size does not match")
# Resize the mask to match the image size
a_resized = Image.fromarray(a).resize(img.size, Image.ANTIALIAS)
a_resized = np.clip(a_resized, 0, 255).astype(np.uint8)
img.putalpha(Image.fromarray(a_resized, mode='L'))
metadata = None
if not args.disable_metadata:
metadata = PngInfo()