CavinHuang-comfyui-nodes-docs/docs/ShuffleChannels.md

Documentation

• Class name: ShuffleChannels
• Category: image/filters
• Output node: False

ShuffleChannels节点旨在操作图像的颜色通道。它允许重新分配颜色通道，包括红色、绿色、蓝色，以及可选的alpha通道，将它们映射到不同的颜色值或常量值。这一功能可用于创造各种视觉效果或纠正图像中的颜色通道错位。

Input types

Required

• image
  • 待处理的输入图像。该图像的颜色通道将根据指定的映射进行重排。
  • Comfy dtype: IMAGE
  • Python dtype: torch.Tensor
• red
  • 指定输出图像中红色通道的映射。它可以映射到另一个颜色通道或设置为常量值。
  • Comfy dtype: COMBO[STRING]
  • Python dtype: str
• green
  • 指定输出图像中绿色通道的映射。它可以映射到另一个颜色通道或设置为常量值。
  • Comfy dtype: COMBO[STRING]
  • Python dtype: str
• blue
  • 指定输出图像中蓝色通道的映射。它可以映射到另一个颜色通道或设置为常量值。
  • Comfy dtype: COMBO[STRING]
  • Python dtype: str
• alpha
  • 可选参数，指定输出图像中alpha（透明度）通道的映射。它可以映射到另一个通道，设置为常量值，或保持不变。
  • Comfy dtype: COMBO[STRING]
  • Python dtype: str

Output types

• image
  • 根据指定映射重排颜色通道后的输出图像。
  • Comfy dtype: IMAGE
  • Python dtype: torch.Tensor

Usage tips

• Infra type: GPU
• Common nodes: unknown

Source code

class ShuffleChannels:
    @classmethod
    def INPUT_TYPES(s):
        return {
            "required": {
                "image": ("IMAGE",),
                "red": (Channel_List, {"default": "red"}),
                "green": (Channel_List, {"default": "green"}),
                "blue": (Channel_List, {"default": "blue"}),
                "alpha": (Alpha_List, {"default": "none"}),
            },
        }

    RETURN_TYPES = ("IMAGE",)
    FUNCTION = "shuffle"

    CATEGORY = "image/filters"

    def shuffle(self, image, red, green, blue, alpha):
        ch = 3 if alpha == "none" else 4
        t = torch.zeros((image.shape[0], image.shape[1], image.shape[2], ch), dtype=image.dtype, device=image.device)
        image_copy = image.detach().clone()
        
        ch_key = [red, green, blue, alpha]
        for i in range(ch):
            if ch_key[i] == "white":
                t[:,:,:,i] = 1
            elif ch_key[i] == "red":
                t[:,:,:,i] = image_copy[:,:,:,0]
            elif ch_key[i] == "green":
                t[:,:,:,i] = image_copy[:,:,:,1]
            elif ch_key[i] == "blue":
                t[:,:,:,i] = image_copy[:,:,:,2]
            elif ch_key[i] == "alpha":
                if image.shape[3] > 3:
                    t[:,:,:,i] = image_copy[:,:,:,3]
                else:
                    t[:,:,:,i] = 1
        
        return(t,)