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

Update image_nodes.py

Browse Source
This commit is contained in:
kijai 2024-11-06 13:01:27 +02:00
parent 3b065864fa
commit 3f903091b3
1 changed files with 1 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
nodes/image_nodes.py
View File

@ -2474,8 +2474,6 @@ class ImageCropByMaskAndResize:
return (int(x0), int(y0), int(w), int(h)) return (int(x0), int(y0), int(w), int(h))
def crop(self, image, mask, base_resolution, padding=0, min_crop_resolution=128, max_crop_resolution=512): def crop(self, image, mask, base_resolution, padding=0, min_crop_resolution=128, max_crop_resolution=512):
print("mask shape: ",mask.shape)
print("image shape: ",image.shape)
image_list = [] image_list = []
mask_list = [] mask_list = []
bbox_list = [] bbox_list = []
@ -2500,9 +2498,6 @@ class ImageCropByMaskAndResize:
cropped_image = common_upscale(cropped_image, target_width, target_height, "lanczos", "disabled") cropped_image = common_upscale(cropped_image, target_width, target_height, "lanczos", "disabled")
cropped_image = cropped_image.movedim(1, -1).squeeze(0) cropped_image = cropped_image.movedim(1, -1).squeeze(0)
print("cropped_image shape: ",cropped_image.shape)
print("cropped_mask shape: ",cropped_mask.shape)
cropped_mask = cropped_mask.unsqueeze(0).unsqueeze(0) cropped_mask = cropped_mask.unsqueeze(0).unsqueeze(0)
cropped_mask = common_upscale(cropped_mask, target_width, target_height, 'bilinear', "disabled") cropped_mask = common_upscale(cropped_mask, target_width, target_height, 'bilinear', "disabled")
cropped_mask = cropped_mask.squeeze(0).squeeze(0) cropped_mask = cropped_mask.squeeze(0).squeeze(0)
@ -2548,7 +2543,7 @@ class ImageUncropByMask:
resized_source = resized_source.movedim(1, -1).squeeze(0) resized_source = resized_source.movedim(1, -1).squeeze(0)
# Resize mask to match the bounding box dimensions # Resize mask to match the bounding box dimensions
resized_mask = F.interpolate(mask[i].unsqueeze(0).unsqueeze(0), size=(bbox_height, bbox_width), mode='bilinear') resized_mask = common_upscale(mask[i].unsqueeze(0).unsqueeze(0), bbox_width, bbox_height, "bilinear", "disabled")
resized_mask = resized_mask.squeeze(0).squeeze(0) resized_mask = resized_mask.squeeze(0).squeeze(0)
# Calculate padding values # Calculate padding values