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

Documentation

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

InsertImageBatchByIndexes节点旨在将特定图像插入到给定的图像批次中的指定索引位置，同时保持批次的原始序列顺序。这个功能对于需要精确操作图像序列的任务特别有用，例如将处理过的图像插回其原始上下文中。

Input types

Required

• images
  • 原始图像批次，新图像将被插入其中。它作为插入操作的基础。
  • Comfy dtype: IMAGE
  • Python dtype: torch.Tensor
• images_to_insert
  • 要插入到原始批次中指定索引位置的图像。这个参数对于定义哪些图像被添加以及添加到哪里至关重要。
  • Comfy dtype: IMAGE
  • Python dtype: torch.Tensor
• insert_indexes
  • 原始批次中新图像应该被插入的索引位置。这个参数决定了插入的位置，将新图像与原始序列对齐。
  • Comfy dtype: INDEXES
  • Python dtype: List[int]

Output types

• images_after_insert
  • 插入操作后更新的图像批次，保留了原始序列顺序，并在指定位置集成了新图像。
  • Comfy dtype: IMAGE
  • Python dtype: torch.Tensor

Usage tips

• Infra type: GPU
• Common nodes: unknown

Source code

class InsertImageBatchByIndexes:

    @classmethod
    def INPUT_TYPES(cls):
        return {
            "required": {
                "images": ("IMAGE",), 
                "images_to_insert": ("IMAGE",), 
                "insert_indexes": ("INDEXES",),
            },
        }

    RETURN_TYPES = ("IMAGE", )
    RETURN_NAMES = ("images_after_insert", )
    FUNCTION = "insert"
    CATEGORY = "KJNodes/image"
    DESCRIPTION = """
This node is designed to be use with node FilterZeroMasksAndCorrespondingImages
It inserts the images_to_insert into images according to insert_indexes

Returns:
    images_after_insert: updated original images with origonal sequence order
"""
    
    def insert(self, images, images_to_insert, insert_indexes):        
        images_after_insert = images
        
        if images_to_insert is not None and insert_indexes is not None:
            images_to_insert_num = len(images_to_insert)
            insert_indexes_num = len(insert_indexes)
            if images_to_insert_num == insert_indexes_num:
                images_after_insert = []

                i_images = 0
                for i in range(len(images) + images_to_insert_num):
                    if i in insert_indexes:
                        images_after_insert.append(images_to_insert[insert_indexes.index(i)])
                    else:
                        images_after_insert.append(images[i_images])
                        i_images += 1
                        
                images_after_insert = torch.stack(images_after_insert, dim=0)
                
            else:
                print(f"[WARNING] skip this node, due to number of images_to_insert ({images_to_insert_num}) is not equal to number of insert_indexes ({insert_indexes_num})")


        return (images_after_insert, )