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

Add image concanate for custom grids

Browse Source
This commit is contained in:
kijai 2023-11-05 11:37:37 +02:00
parent 2a7b1ce405
commit 179c462e9b
1 changed files with 31 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
nodes.py
View File

@ -1100,7 +1100,32 @@ class SaveImageWithAlpha:
return { "ui": { "images": results } }
class ImageConcanate:
@classmethod
def INPUT_TYPES(s):
return {"required": {
"image1": ("IMAGE",),
"image2": ("IMAGE",),
"direction": (
[ 'right',
'down',
],
{
"default": 'right'
}),
}}
RETURN_TYPES = ("IMAGE",)
FUNCTION = "concanate"
CATEGORY = "KJNodes"
def concanate(self, image1, image2, direction):
if direction == 'right':
row = torch.cat((image1, image2), dim=2)
elif direction == 'down':
row = torch.cat((image1, image2), dim=1)
return (row,)
class ImageGridComposite2x2:
@classmethod
def INPUT_TYPES(s):
@ -1108,8 +1133,7 @@ class ImageGridComposite2x2:
"image1": ("IMAGE",),
"image2": ("IMAGE",),
"image3": ("IMAGE",),
"image4": ("IMAGE",),
"image4": ("IMAGE",),
}}
RETURN_TYPES = ("IMAGE",)
@ -1134,8 +1158,7 @@ class ImageGridComposite3x3:
"image6": ("IMAGE",),
"image7": ("IMAGE",),
"image8": ("IMAGE",),
"image9": ("IMAGE",),
"image9": ("IMAGE",),
}}
RETURN_TYPES = ("IMAGE",)
@ -1173,7 +1196,8 @@ NODE_CLASS_MAPPINGS = {
"SaveImageWithAlpha": SaveImageWithAlpha,
"ReverseImageBatch": ReverseImageBatch,
"ImageGridComposite2x2": ImageGridComposite2x2,
"ImageGridComposite3x3": ImageGridComposite3x3
"ImageGridComposite3x3": ImageGridComposite3x3,
"ImageConcanate": ImageConcanate
}
NODE_DISPLAY_NAME_MAPPINGS = {
"INTConstant": "INT Constant",
@ -1198,5 +1222,6 @@ NODE_DISPLAY_NAME_MAPPINGS = {
"SaveImageWithAlpha": "SaveImageWithAlpha",
"ReverseImageBatch": "ReverseImageBatch",
"ImageGridComposite2x2": "ImageGridComposite2x2",
"ImageGridComposite3x3": "ImageGridComposite3x3"
"ImageGridComposite3x3": "ImageGridComposite3x3",
"ImageConcanate": "ImageConcanate"
}