68 lines
2.0 KiB
Python
68 lines
2.0 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Filename: dynamic_image_input.py
|
|
# Developer: jammyfu
|
|
# Category: 🎨Painting👓Coder/🖼️Image
|
|
|
|
import torch
|
|
|
|
class DynamicImageCombiner:
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
return {
|
|
"required": {
|
|
"image_1": ("IMAGE",), # 添加一个初始图像输入
|
|
},
|
|
"optional": {},
|
|
"_meta": {
|
|
"preferred_width": 300,
|
|
"maintain_dimensions": True
|
|
}
|
|
}
|
|
|
|
RETURN_TYPES = ("IMAGE",)
|
|
RETURN_NAMES = ("images",)
|
|
OUTPUT_IS_LIST = (True,)
|
|
FUNCTION = "combine_images"
|
|
CATEGORY = "🎨Painting👓Coder/🖼️Image"
|
|
|
|
def combine_images(self, **kwargs):
|
|
try:
|
|
images = []
|
|
# 获取所有图像输入键
|
|
image_inputs = sorted(
|
|
[k for k in kwargs.keys() if k.startswith('image_')],
|
|
key=lambda x: int(x.split('_')[1])
|
|
)
|
|
|
|
# 只处理非None的输入（已连接的输入）
|
|
for key in image_inputs:
|
|
input_image = kwargs.get(key)
|
|
|
|
# 跳过未连接的输入（None值）
|
|
if input_image is None:
|
|
continue
|
|
|
|
# 处理已连接的输入
|
|
if isinstance(input_image, torch.Tensor):
|
|
images.append(input_image)
|
|
|
|
# 如果没有有效的图像输入，返回空图像
|
|
if not images:
|
|
empty_image = torch.zeros((1, 512, 512, 3))
|
|
return ([empty_image],)
|
|
|
|
return (images,)
|
|
|
|
except Exception as e:
|
|
print(f"Error in DynamicImageCombiner: {str(e)}")
|
|
empty_image = torch.zeros((1, 512, 512, 3))
|
|
return ([empty_image],)
|
|
|
|
# 更新节点映射
|
|
NODE_CLASS_MAPPINGS = {
|
|
"PaintingCoder::DynamicImageCombiner": DynamicImageCombiner,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"PaintingCoder::DynamicImageCombiner": "Dynamic Image Input 🖼️",
|
|
} |