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

Documentation

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

从图片中获取主颜色或平均色。

Input types

Required

• image

  • 输入的图像。
  • Comfy dtype: IMAGE
  • Python dtype: torch.Tensor

• mode

  • 模式，有两种可选择，主颜色main_color和平均色average。
  • Comfy dtype: STRING
  • Python dtype: str
  • 可选值: 'average', 'main'

Output types

• RGB color in HEX

  • 使用16进制RGB字符串格式描述，例如 '#FA3D86'。
  • Comfy dtype: STRING
  • Python dtype: str

• HSV color in list

  • 使用HSV格式描述的颜色。
  • Comfy dtype: LIST
  • Python dtype: list

Usage tips

• Infra type: GPU

Source code

class GetColorTone:

    def __init__(self):
        pass

    @classmethod
    def INPUT_TYPES(self):
        mode_list = ['main_color', 'average']
        return {
            "required": {
                "image": ("IMAGE", ),  #
                "mode": (mode_list,),  # 主色/平均色
            },
            "optional": {
            }
        }

    RETURN_TYPES = ("STRING", "LIST")
    RETURN_NAMES = ("RGB color in HEX", "HSV color in list")
    FUNCTION = 'get_color_tone'
    CATEGORY = '😺dzNodes/LayerUtility'

    def get_color_tone(self, image, mode,):
        if image.shape[0] > 0:
            image = torch.unsqueeze(image[0], 0)
        _canvas = tensor2pil(image).convert('RGB')
        _canvas = gaussian_blur(_canvas, int((_canvas.width + _canvas.height) / 200))
        if mode == 'main_color':
            ret_color = get_image_color_tone(_canvas)
        else:
            ret_color = get_image_color_average(_canvas)
        hsv_color = RGB_to_HSV(Hex_to_RGB(ret_color))

        return (ret_color, hsv_color)