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

Merge pull request #375 from asaf92/fix-image-pad-outpaint-mask-scaling

Browse Source 
Fix ImagePadForOutpaintTargetSize mask scaling bug
This commit is contained in:
Jukka Seppänen 2025-09-16 18:15:28 +03:00 committed by GitHub
commit af20cb3333
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
nodes/image_nodes.py
View File

@ -1128,16 +1128,17 @@ class ImagePadForOutpaintTargetSize:
# Downscale the image
image_scaled = common_upscale(image, new_width, new_height, upscale_method, "disabled").movedim(1,-1)
if mask is not None:
mask_scaled = mask.unsqueeze(0) # Add an extra dimension for batch size
mask_scaled = F.interpolate(mask_scaled, size=(new_height, new_width), mode="nearest")
mask_scaled = mask_scaled.squeeze(0) # Remove the extra dimension after interpolation
else:
mask_scaled = mask
else:
# If downscaling is not needed, use the original image dimensions
image_scaled = image
mask_scaled = mask
# Ensure mask dimensions match image dimensions
if mask is not None:
mask_scaled = mask.unsqueeze(0) # Add an extra dimension for batch size
mask_scaled = F.interpolate(mask_scaled, size=(new_height, new_width), mode="nearest")
mask_scaled = mask_scaled.squeeze(0) # Remove the extra dimension after interpolation
else:
mask_scaled = None
# Calculate how much padding is needed to reach the target dimensions
pad_top = max(0, (target_height - new_height) // 2)
@ -3867,4 +3868,4 @@ class LoadVideosFromFolder:
def IS_CHANGED(s, video, **kwargs):
if s.vhs_nodes is not None:
return s.vhs_nodes.utils.hash_path(video)
return None
return None