xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2025-12-10 21:34:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'main' of https://github.com/kijai/ComfyUI-KJNodes

Browse Source
This commit is contained in:
kijai 2024-05-13 22:46:49 +03:00
commit 68471f65b3
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
nodes/mask_nodes.py
View File

@ -1195,10 +1195,9 @@ Resizes the mask or batch of masks to the specified width and height.
ratio = min(width / ow, height / oh) ratio = min(width / ow, height / oh)
width = round(ow*ratio) width = round(ow*ratio)
height = round(oh*ratio) height = round(oh*ratio)
outputs = mask.unsqueeze(1)
outputs = mask.unsqueeze(0) # Add an extra dimension for batch size
outputs = F.interpolate(outputs, size=(height, width), mode="nearest") outputs = F.interpolate(outputs, size=(height, width), mode="nearest")
outputs = outputs.squeeze(0) # Remove the extra dimension after interpolation outputs = outputs.squeeze(1)
return(outputs, outputs.shape[2], outputs.shape[1],) return(outputs, outputs.shape[2], outputs.shape[1],)