ImageBatchToList

Documentation

• Class name: ImageBatchToList
• Category: Bmad/image
• Output node: False

This node is designed to convert a batch of images into a list format, facilitating operations that require individual image processing or manipulation within a batch context.

Input types

Required

• images
  • Represents the batch of images to be converted into a list. This input is crucial for enabling the node to iterate over each image in the batch and perform the conversion.
  • Comfy dtype: IMAGE
  • Python dtype: torch.Tensor

Output types

• image
  • Comfy dtype: IMAGE
  • The output is a list of images, where each image in the batch has been separated and is now an individual element in the list.
  • Python dtype: List[torch.Tensor]

Usage tips

• Infra type: GPU
• Common nodes: unknown

Source code

class ImageBatchToList:
    @classmethod
    def INPUT_TYPES(s):
        return {"required": {"images": ("IMAGE",)}}

    RETURN_TYPES = ("IMAGE",)
    FUNCTION = "to_list"
    CATEGORY = "Bmad/image"
    OUTPUT_IS_LIST = (True,)

    def to_list(self, images):
        image_list = [images[i][None, ...] for i in range(images.shape[0])]
        return (image_list,)