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

Documentation

• Class name: ImageSizeAndBatchSize
• Category: Animate Diff/Utils
• Output node: False
• Repo Ref: https://github.com/ArtVentureX/comfyui-animatediff.git

此类节点旨在提取并返回关键的图像属性，如大小和批量维度，便于系统内进行进一步的处理和分析。

Input types

Required

• image
  • 图像参数至关重要，因为它是节点执行功能的主输入。它通过确定输出的尺寸和批量大小来影响节点的操作。
  • Comfy dtype: IMAGE
  • Python dtype: torch.Tensor

Output types

• width
  • 宽度参数代表图像的水平维度，是图像分析和处理的一个基本方面。
  • Comfy dtype: INT
  • Python dtype: int
• height
  • 高度参数表示图像的垂直维度，在结构理解和操作图像数据中起着关键作用。
  • Comfy dtype: INT
  • Python dtype: int
• batch_size
  • 批量大小参数指示一批图像中图像的数量，这对于优化计算效率和在图像处理期间管理资源至关重要。
  • Comfy dtype: INT
  • Python dtype: int

Usage tips

• Infra type: CPU

Source code

class ImageSizeAndBatchSize:

    @classmethod
    def INPUT_TYPES(s):
        return {'required': {'image': ('IMAGE',)}}
    CATEGORY = 'Animate Diff/Utils'
    RETURN_TYPES = ('INT', 'INT', 'INT')
    RETURN_NAMES = ('width', 'height', 'batch_size')
    FUNCTION = 'batch_size'

    def batch_size(self, image: Tensor):
        (batch_size, height, width) = image.shape[0:3]
        return (width, height, batch_size)