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

Documentation

• Class name: ImageExtractChannel
• Category: Art Venture/Utils
• Output node: False

ImageExtractChannel节点专门用于从给定的一组图像中提取特定的颜色通道（红色、绿色、蓝色或阿尔法）。它允许对单个颜色通道进行操作和分析，这对于各种图像处理任务至关重要，如创建蒙版或隔离颜色组件。

Input types

Required

• images
  • 将从中提取特定通道的图像。这个输入对于确定要处理的源图像至关重要。
  • Comfy dtype: IMAGE
  • Python dtype: torch.Tensor
• channel
  • 指定要从图像中提取的颜色通道（红色、绿色、蓝色或阿尔法）。这个选择通过隔离所需的通道直接影响输出。
  • Comfy dtype: COMBO[STRING]
  • Python dtype: str

Output types

• channel_data
  • 从输入图像中提取的通道数据，以蒙版形式提供。这个输出对于进一步的图像处理或分析任务非常有用。
  • Comfy dtype: MASK
  • Python dtype: torch.Tensor

Usage tips

• Infra type: GPU
• Common nodes: unknown

Source code

class UtilImageExtractChannel:
    @classmethod
    def INPUT_TYPES(s):
        return {
            "required": {
                "images": ("IMAGE",),
                "channel": (["R", "G", "B", "A"],),
            }
        }

    RETURN_TYPES = ("MASK",)
    RETURN_NAMES = ("channel_data",)
    CATEGORY = "Art Venture/Utils"
    FUNCTION = "image_extract_alpha"

    def image_extract_alpha(self, images: torch.Tensor, channel):
        # images in shape (N, H, W, C)

        if len(images.shape) < 4:
            images = images.unsqueeze(3).repeat(1, 1, 1, 3)

        if channel == "A" and images.shape[3] < 4:
            raise Exception("Image does not have an alpha channel")

        channel_index = ["R", "G", "B", "A"].index(channel)
        mask = images[:, :, :, channel_index].cpu().clone()

        return (mask,)