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

Documentation

• Class name: ImageBatchRepeatInterleaving
• Category: KJNodes/image
• Output node: False

ImageBatchRepeatInterleaving节点旨在通过重复批次中的每张图像指定次数来扩展图像批次。这个操作通过复制图像有效地增加了批次大小，这对于数据增强或确保统一的批处理大小非常有用。

Input types

Required

• images
  • images参数代表要重复的图像批次。它对于定义将进行重复处理的输入数据至关重要。
  • Comfy dtype: IMAGE
  • Python dtype: torch.Tensor
• repeats
  • repeats参数指定批次中每张图像应重复的次数。它通过确定每张图像的复制因子直接影响最终的批次大小。
  • Comfy dtype: INT
  • Python dtype: int

Output types

• image
  • 输出是一个新的图像批次，其中每张原始图像已根据repeats参数重复，有效地扩大了批次大小。
  • Comfy dtype: IMAGE
  • Python dtype: torch.Tensor

Usage tips

• Infra type: GPU
• Common nodes: unknown

Source code

class ImageBatchRepeatInterleaving:
    
    RETURN_TYPES = ("IMAGE",)
    FUNCTION = "repeat"
    CATEGORY = "KJNodes/image"
    DESCRIPTION = """
Repeats each image in a batch by the specified number of times.  
Example batch of 5 images: 0, 1 ,2, 3, 4  
with repeats 2 becomes batch of 10 images: 0, 0, 1, 1, 2, 2, 3, 3, 4, 4  
"""

    @classmethod
    def INPUT_TYPES(s):
        return {
            "required": {
                 "images": ("IMAGE",),
                 "repeats": ("INT", {"default": 1, "min": 1, "max": 4096}),
        },
    } 
    
    def repeat(self, images, repeats):
       
        repeated_images = torch.repeat_interleave(images, repeats=repeats, dim=0)
        return (repeated_images, )