diff --git a/nodes/image_nodes.py b/nodes/image_nodes.py index 8de598e..bc001da 100644 --- a/nodes/image_nodes.py +++ b/nodes/image_nodes.py @@ -785,8 +785,8 @@ class GetImageSizeAndCount: "image": ("IMAGE",), }} - RETURN_TYPES = ("IMAGE","INT", "INT", "INT",) - RETURN_NAMES = ("image", "width", "height", "count",) + RETURN_TYPES = ("IMAGE","INT", "INT", "INT", "INT",) + RETURN_NAMES = ("image", "width", "height", "count", "channels",) FUNCTION = "getsize" CATEGORY = "KJNodes/image" DESCRIPTION = """ @@ -799,9 +799,10 @@ and passes it through unchanged. width = image.shape[2] height = image.shape[1] count = image.shape[0] + channels = image.shape[3] return {"ui": { - "text": [f"{count}x{width}x{height}"]}, - "result": (image, width, height, count) + "text": [f"{count}x{width}x{height}x{channels}"]}, + "result": (image, width, height, count, channels) } class GetLatentSizeAndCount: diff --git a/web/js/jsnodes.js b/web/js/jsnodes.js index c391bd5..e9998df 100644 --- a/web/js/jsnodes.js +++ b/web/js/jsnodes.js @@ -175,6 +175,7 @@ app.registerExtension({ this.outputs[1]["label"] = "width" this.outputs[2]["label"] = "height" this.outputs[3]["label"] = "count" + this.outputs[4]["label"] = "channels" return v; } //const onGetImageSizeExecuted = nodeType.prototype.onExecuted; @@ -187,6 +188,7 @@ app.registerExtension({ this.outputs[1]["label"] = values[1] + " width" this.outputs[2]["label"] = values[2] + " height" this.outputs[3]["label"] = values[0] + " count" + this.outputs[4]["label"] = values[3] + " channels" return r } break;