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

Documentation

• Class name: SV_random
• Category: Clothing - SAL-VTON
• Output node: False

RandomImageFromDir节点旨在从指定目录中随机选择一张图片。这个功能特别适用于需要图片选择变化性和不可预测性的应用场景，例如数据增强过程或在给定上下文中模拟多样化场景。

Input types

Required

• folder_path
  • 指定将从中随机选择图片的目录路径。这个参数对于确定可供随机选择的图片池至关重要。
  • Comfy dtype: STRING
  • Python dtype: str

Output types

• image
  • 返回从指定目录中随机选择的图片。这个输出对于在进一步处理或显示中引入变化性和随机性至关重要。
  • Comfy dtype: IMAGE
  • Python dtype: torch.Tensor
• mask
  • 提供与随机选择的图片相关联的蒙版（如果适用），以便于进一步的图像处理任务。
  • Comfy dtype: MASK
  • Python dtype: torch.Tensor

Usage tips

• Infra type: CPU
• Common nodes: unknown

Source code

class RandomImageFromDir:
    def __init__(self):
        pass

    @classmethod
    def INPUT_TYPES(cls):
        return {
            "required": {
                "folder_path": ("STRING", {
                    "multiline": False,
                    "default": "./input"
                })
            }
        }

    RETURN_TYPES = ("IMAGE", "MASK")
    FUNCTION = "load_image"
    CATEGORY = node_category

    def load_image(self, folder_path):
        files = os.listdir(folder_path)
        image_extensions = {".jpg", ".jpeg", ".png", ".bmp", ".tiff", ".ico", ".jfif"}
        images = [file for file in files if os.path.splitext(file)[1].lower() in image_extensions]

        random_image = os.path.join(folder_path, random.choice(images))
        img = Image.open(random_image)
        output_images = []
        output_masks = []
        # this is from load_image node. Lazy but works :')
        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):
        return float("NaN")