From 9680b343d7995d1adf7df4596019ef91a5cbe84a Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Thu, 9 Nov 2023 00:26:09 +0200
Subject: [PATCH] Update nodes.py

---
 nodes.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/nodes.py b/nodes.py
index a68baa1..a7f9a21 100644
--- a/nodes.py
+++ b/nodes.py
@@ -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):