From 871749c20842dbdd7696ba77c78ee7d4c246ef6a Mon Sep 17 00:00:00 2001
From: comfyanonymous <121283862+comfyanonymous@users.noreply.github.com>
Date: Wed, 4 Jun 2025 06:40:21 -0700
Subject: [PATCH] Add batch to GetImageSize node. (#8419)

---
 comfy_extras/nodes_images.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/comfy_extras/nodes_images.py b/comfy_extras/nodes_images.py
index 58b29f9a..b1e0d466 100644
--- a/comfy_extras/nodes_images.py
+++ b/comfy_extras/nodes_images.py
@@ -505,8 +505,8 @@ class GetImageSize:
             }
         }
 
-    RETURN_TYPES = (IO.INT, IO.INT)
-    RETURN_NAMES = ("width", "height")
+    RETURN_TYPES = (IO.INT, IO.INT, IO.INT)
+    RETURN_NAMES = ("width", "height", "batch_size")
     FUNCTION = "get_size"
 
     CATEGORY = "image"
@@ -515,12 +515,13 @@ class GetImageSize:
     def get_size(self, image, unique_id=None) -> tuple[int, int]:
         height = image.shape[1]
         width = image.shape[2]
+        batch_size = image.shape[0]
 
         # Send progress text to display size on the node
         if unique_id:
-            PromptServer.instance.send_progress_text(f"width: {width}, height: {height}", unique_id)
+            PromptServer.instance.send_progress_text(f"width: {width}, height: {height}\n batch size: {batch_size}", unique_id)
 
-        return width, height
+        return width, height, batch_size
 
 NODE_CLASS_MAPPINGS = {
     "ImageCrop": ImageCrop,