CavinHuang-comfyui-nodes-docs/docs/IG Load Image.md

Documentation

• Class name: IG Load Image
• Category: 🐓 IG Nodes/IO
• Output node: False

IG Load Image节点旨在从指定目录加载图像，提供了图像预处理功能，如方向校正、转换为特定颜色模式和标准化。它简化了图像加载和预处理的复杂性，使图像数据更容易集成到工作流程中。

Input types

Required

• image_path
  • image_path参数指定要加载的图像的路径。它在确定将要处理哪个图像文件方面起着关键作用，影响节点的执行和最终的图像数据。
  • Comfy dtype: STRING
  • Python dtype: str

Output types

• image
  • image参数代表经过预处理的图像数据，可用于进一步处理或分析。
  • Comfy dtype: IMAGE
  • Python dtype: torch.Tensor
• mask
  • mask参数为加载的图像提供了一个掩码，对于需要分割或特定区域处理的操作很有用。
  • Comfy dtype: MASK
  • Python dtype: torch.Tensor

Usage tips

• Infra type: CPU
• Common nodes: unknown

Source code

class IG_LoadImage:
    @classmethod
    def INPUT_TYPES(s):
        return {
            "required": {
                "image_path": ("STRING", {"forceInput": True}),
            },
        }
    
    RETURN_TYPES = ("IMAGE", "MASK")
    FUNCTION = "main"

    CATEGORY = TREE_IO

    def main(self, image_path: str, **kwargs):
        img = Image.open(image_path)
        output_images = []
        output_masks = []
        for i in ImageSequence.Iterator(img):
            i = ImageOps.exif_transpose(i)
            if i.mode == 'I':
                i = i.point(lambda i: i * (1 / 255))
            image = i.convert("RGB")
            image = np.array(image).astype(np.float32) / 255.0
            image = torch.from_numpy(image)[None,]
            if 'A' in i.getbands():
                mask = np.array(i.getchannel('A')).astype(np.float32) / 255.0
                mask = 1. - torch.from_numpy(mask)
            else:
                mask = torch.zeros((64,64), dtype=torch.float32, device="cpu")
            output_images.append(image)
            output_masks.append(mask.unsqueeze(0))

        if len(output_images) > 1:
            output_image = torch.cat(output_images, dim=0)
            output_mask = torch.cat(output_masks, dim=0)
        else:
            output_image = output_images[0]
            output_mask = output_masks[0]

        return (output_image, output_mask)
    
    @classmethod
    def IS_CHANGED(s, image_path: str, **kwargs):
        return is_changed_image(image_path, **kwargs)