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

Update image_nodes.py

Browse Source
This commit is contained in:
kijai 2024-10-26 22:31:28 +03:00
parent 19ec49ae2b
commit f59e410568
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
nodes/image_nodes.py
View File

@ -2484,11 +2484,15 @@ class ImageCropByMaskAndResize:
target_height = base_resolution
target_width = int(base_resolution * aspect_ratio)
cropped_image = F.interpolate(cropped_image, size=(target_height, target_width), mode='bilinear', align_corners=False)
# Ensure dimensions are divisible by 8
target_width = (target_width + 7) // 8 * 8
target_height = (target_height + 7) // 8 * 8
cropped_image = common_upscale(cropped_image, target_width, target_height, "lanczos", "disabled")
cropped_image = cropped_image.movedim(1, -1).squeeze(0)
cropped_mask = cropped_mask.unsqueeze(0).unsqueeze(0)
cropped_mask = F.interpolate(cropped_mask, size=(target_height, target_width), mode='nearest')
cropped_mask = F.interpolate(cropped_mask, size=(target_height, target_width), mode='bilinear')
cropped_mask = cropped_mask.squeeze(0).squeeze(0)
image_list.append(cropped_image)