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

Documentation

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

InsertImagesToBatchIndexed 节点旨在将指定的图像插入到原始图像批次的特定位置，从而有效地通过在指定位置添加新图像来修改原始序列。这一操作允许动态操作图像批次，使得用户能够根据各种应用需求自定义图像序列。

Input types

Required

• original_images
  • 代表将插入新图像的原始图像批次。它对于定义将被修改的基础序列至关重要。
  • Comfy dtype: IMAGE
  • Python dtype: torch.Tensor
• images_to_insert
  • 指定要插入原始批次的图像。此参数对于确定哪些新图像将被添加到序列中至关重要。
  • Comfy dtype: IMAGE
  • Python dtype: torch.Tensor
• indexes
  • 一个由逗号分隔的索引字符串，指示新图像应该插入原始批次的位置。此参数决定了图像将被添加的位置，从而影响批次的最终排列。
  • Comfy dtype: 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 InsertImagesToBatchIndexed:
    
    RETURN_TYPES = ("IMAGE",)
    FUNCTION = "insertimagesfrombatch"
    CATEGORY = "KJNodes/image"
    DESCRIPTION = """
Inserts images at the specified indices into the original image batch.
"""

    @classmethod
    def INPUT_TYPES(s):
        return {
            "required": {
                "original_images": ("IMAGE",),
                "images_to_insert": ("IMAGE",),
                "indexes": ("STRING", {"default": "0, 1, 2", "multiline": True}),
            },
        }
    
    def insertimagesfrombatch(self, original_images, images_to_insert, indexes):
        
        # Parse the indexes string into a list of integers
        index_list = [int(index.strip()) for index in indexes.split(',')]
        
        # Convert list of indices to a PyTorch tensor
        indices_tensor = torch.tensor(index_list, dtype=torch.long)
        
        # Ensure the images_to_insert is a tensor
        if not isinstance(images_to_insert, torch.Tensor):
            images_to_insert = torch.tensor(images_to_insert)
        
        # Insert the images at the specified indices
        for index, image in zip(indices_tensor, images_to_insert):
            original_images[index] = image
        
        return (original_images,)