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

Update nodes.py

Browse Source
This commit is contained in:
kijai 2023-11-09 00:26:09 +02:00
parent dbd2ea2406
commit 9680b343d7
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
nodes.py
View File

@ -1289,11 +1289,15 @@ class BatchCropFromMask:
"IMAGE", "IMAGE",
"IMAGE", "IMAGE",
"BBOX", "BBOX",
"INT",
"INT",
) )
RETURN_NAMES = ( RETURN_NAMES = (
"original_images", "original_images",
"cropped_images", "cropped_images",
"bboxes", "bboxes",
"width",
"height",
) )
FUNCTION = "crop" FUNCTION = "crop"
CATEGORY = "KJNodes/masking" CATEGORY = "KJNodes/masking"
@ -1343,7 +1347,7 @@ class BatchCropFromMask:
cropped_images.append(cropped_img) cropped_images.append(cropped_img)
cropped_out = torch.stack(cropped_images, dim=0) cropped_out = torch.stack(cropped_images, dim=0)
return (original_images, cropped_out, bounding_boxes,) return (original_images, cropped_out, bounding_boxes, bbox_size, bbox_size,)
def bbox_to_region(bbox, target_size=None): def bbox_to_region(bbox, target_size=None):