CavinHuang-comfyui-nodes-docs/docs/LayerUtility%3A ColorImageV2.md

Documentation

• Class name: ColorImage
• Category: 😺dzNodes/LayerUtility
• Output node: False
• Repo Ref: https://github.com/chflame163/ComfyUI_LayerStyle

生成一张指定颜色和大小的图片。ColorImage的V2升级版。

Input types

Required

• size

  • 尺寸预设。预设可以用户自定义。如果有size_as输入，此处选项将被忽略。
  • Comfy dtype: STRING
  • Python dtype: str
  • Options: {"default": "custom", "options": ["custom", "512x512", "1024x1024", "2048x2048", "4096x4096", "8192x8192", "16384x16384", "32768x32768"]}

• custom_width

  • 图像宽度。当size设置为"custom"时有效。如果有size_as输入，此处选项将被忽略。
  • Comfy dtype: INT
  • Python dtype: int
  • Options: {"default": 512, "min": 4, "max": 99999, "step": 1}

• custom_height

  • 图像高度。当size设置为"custom"时有效。如果有size_as输入，此处选项将被忽略。
  • Comfy dtype: INT
  • Python dtype: int
  • Options: {"default": 512, "min": 4, "max": 99999, "step": 1}

• color

  • 图片的颜色。
  • Comfy dtype: STRING
  • Python dtype: str
  • Options: {"default": "#000000"}

Optional

• size_as
  • 作为参考的图片。如果设置了此项，将会忽略size选项。
  • Comfy dtype: IMAGE
  • Python dtype: torch.Tensor

Output types

• image
  • 输出的图片。
  • Comfy dtype: IMAGE
  • Python dtype: torch.Tensor

Usage tips

• Infra type: GPU

Source code

class ColorImageV2:

    def __init__(self):
        pass

    @classmethod
    def INPUT_TYPES(self):
        size_list = ['custom']
        size_list.extend(load_custom_size())
        return {
            "required": {
                "size": (size_list,),
                "custom_width": ("INT", {"default": 512, "min": 4, "max": 99999, "step": 1}),
                "custom_height": ("INT", {"default": 512, "min": 4, "max": 99999, "step": 1}),
                "color": ("STRING", {"default": "#000000"},),
            },
            "optional": {
                "size_as": (any, {}),
            }
        }

    RETURN_TYPES = ("IMAGE", )
    RETURN_NAMES = ("image", )
    FUNCTION = 'color_image_v2'
    CATEGORY = '😺dzNodes/LayerUtility'

    def color_image_v2(self, size, custom_width, custom_height, color, size_as=None ):

        if size_as is not None:
            if size_as.shape[0] > 0:
                _asimage = tensor2pil(size_as[0])
            else:
                _asimage = tensor2pil(size_as)
            width, height = _asimage.size
        else:
            if size == 'custom':
                width = custom_width
                height = custom_height
            else:
                try:
                    _s = size.split('x')
                    width = int(_s[0].strip())
                    height = int(_s[1].strip())
                except Exception as e:
                    log(f"Warning: {NODE_NAME} invalid size, check {custom_size_file}", message_type='warning')
                    width = custom_width
                    height = custom_height

        ret_image = Image.new('RGB', (width, height), color=color)
        return (pil2tensor(ret_image), )