CavinHuang-comfyui-nodes-docs/docs/Get First Image.md

Documentation

• Class name: GetFirstImage
• Category: DragNUWA
• Output node: False
• Repo Ref: https://github.com/chaojie/ComfyUI-DragNUWA.git

该节点旨在从一批图像中提取特定的子集，专注于选择过程并确保所需图像被隔离以供进一步处理。

Input types

Required

• images
  • 图像参数至关重要，因为它是节点操作的主要输入。预期它是一个包含一批图像的张量，节点将根据定义的标准从这些图像中选择特定的图像。
  • Comfy dtype: COMBO[torch.Tensor]
  • Python dtype: torch.Tensor

Output types

• images
  • 该节点的输出是根据指定标准选择的输入图像的精炼子集。这个子集旨在进行进一步的分析或处理。
  • Comfy dtype: COMBO[torch.Tensor]
  • Python dtype: torch.Tensor

Usage tips

• Infra type: CPU

Source code

class GetFirstImage:

    @classmethod
    def INPUT_TYPES(s):
        return {'required': {'images': ('IMAGE',)}}
    RETURN_TYPES = ('IMAGE',)
    FUNCTION = 'run'
    OUTPUT_NODE = False
    CATEGORY = 'DragNUWA'

    def run(self, images: torch.Tensor):
        shape = images.shape
        len_first_dim = shape[0]
        selected_indexes = f'0'
        selected_index: list[int] = []
        total_indexes: list[int] = list(range(len_first_dim))
        for s in selected_indexes.strip().split(','):
            try:
                if ':' in s:
                    _li = s.strip().split(':', maxsplit=1)
                    _start = _li[0]
                    _end = _li[1]
                    if _start and _end:
                        selected_index.extend(total_indexes[int(_start):int(_end)])
                    elif _start:
                        selected_index.extend(total_indexes[int(_start):])
                    elif _end:
                        selected_index.extend(total_indexes[:int(_end)])
                else:
                    x: int = int(s.strip())
                    if x < len_first_dim:
                        selected_index.append(x)
            except:
                pass
        return (images[selected_index, :, :, :],)