xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2025-12-09 21:04:41 +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",
"BBOX",
"INT",
"INT",
)
RETURN_NAMES = (
"original_images",
"cropped_images",
"bboxes",
"width",
"height",
)
FUNCTION = "crop"
CATEGORY = "KJNodes/masking"
@ -1343,7 +1347,7 @@ class BatchCropFromMask:
cropped_images.append(cropped_img)
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):