48 lines
1.5 KiB
Markdown
48 lines
1.5 KiB
Markdown
# 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)
|
|
``` |