xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2026-04-16 03:27:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add channels to GetImageSizeAndCount

Browse Source
This commit is contained in:
MakkiShizu 2025-04-24 18:59:53 +08:00
parent 086ac449a2
commit 6717a58837
2 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
nodes/image_nodes.py
View File

@ -766,8 +766,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 = """
@ -780,9 +780,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 ImageBatchRepeatInterleaving:

2
web/js/jsnodes.js
View File

@ -174,6 +174,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;
@ -186,6 +187,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;