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

Documentation

• Class name: MakeImageBatch
• Category: ImpactPack/Util
• Output node: False
• Repo Ref: https://github.com/ltdrdata/ComfyUI-Impact-Pack.git

‘MakeImageBatch’节点旨在高效地将多个图像组合成一个单一的批次以供进一步处理。它通过在必要时调整大小以确保批次中的所有图像具有相同的尺寸，并将它们沿批次维度连接起来。该节点在准备图像处理任务的统一数据集方面发挥着关键作用，例如批量神经网络训练或批量图像操作。

Input types

Required

• image1
  • ‘image1’参数是用作图像批次尺寸参考的第一个图像。它至关重要，因为如果后续图像的尺寸不同，它们将被调整以匹配其尺寸。节点的功能依赖于此参数来创建用于下游任务的统一批次。
  • Comfy dtype: IMAGE
  • Python dtype: torch.Tensor

Optional

• kwargs
  • ‘kwargs’参数允许在批次中包含额外的图像。通过此参数提供的每个图像都以与‘image1’相同的方式进行处理，确保批次的一致性。‘kwargs’的灵活性使节点能够适应不同场景中的可变数量的图像，增强了其实用性。
  • Comfy dtype: COMBO[str, IMAGE]
  • Python dtype: Dict[str, torch.Tensor]

Output types

• image_batch
  • ‘image_batch’输出是已连接的图像批次，这些图像已被调整大小以保持批次内的一致性。这个输出很重要，因为它为后续的图像处理任务奠定了基础，确保批次准备好进行如神经网络输入或批量图像编辑等操作。
  • Comfy dtype: IMAGE
  • Python dtype: torch.Tensor

Usage tips

• Infra type: CPU

Source code

class MakeImageBatch:

    @classmethod
    def INPUT_TYPES(s):
        return {'required': {'image1': ('IMAGE',)}}
    RETURN_TYPES = ('IMAGE',)
    FUNCTION = 'doit'
    CATEGORY = 'ImpactPack/Util'

    def doit(self, **kwargs):
        image1 = kwargs['image1']
        del kwargs['image1']
        images = [value for value in kwargs.values()]
        if len(images) == 0:
            return (image1,)
        else:
            for image2 in images:
                if image1.shape[1:] != image2.shape[1:]:
                    image2 = comfy.utils.common_upscale(image2.movedim(-1, 1), image1.shape[2], image1.shape[1], 'lanczos', 'center').movedim(1, -1)
                image1 = torch.cat((image1, image2), dim=0)
            return (image1,)