Files
CavinHuang-comfyui-nodes-docs/docs/easy imageSave.md
2024-05-26 20:35:16 +08:00

2.1 KiB

Documentation

该节点简化了图像数据的保存过程,提供了一种直接保存视觉输出的机制。它支持可选的图像压缩和预览生成,确保图像能够方便地用于后续使用或检查。

Input types

Required

  • images
    • images参数是必需的,因为它携带了需要保存的视觉数据。它通过决定存储的内容和输出的外观来影响整个操作。
    • Comfy dtype: IMAGE
    • Python dtype: PIL.Image or torch.Tensor

Optional

  • filename_prefix
    • filename_prefix用于预定义保存图像的命名约定,这对于在输出目录中组织和识别文件至关重要。
    • Comfy dtype: STRING
    • Python dtype: str
  • only_preview
    • only_preview参数控制是否生成并保存图像的预览。它通过决定生成的输出类型来影响节点的操作。
    • Comfy dtype: BOOLEAN
    • Python dtype: bool

Output types

  • None
    • 这个节点的输出不是一个数据对象,而是保存图像的动作。没有返回值表示保存过程已经完成。
    • Comfy dtype: None
    • Python dtype: None

Usage tips

  • Infra type: CPU

Source code

class imageSaveSimple:

    def __init__(self):
        self.output_dir = folder_paths.get_output_directory()
        self.type = 'output'
        self.prefix_append = ''
        self.compress_level = 4

    @classmethod
    def INPUT_TYPES(s):
        return {'required': {'images': ('IMAGE',), 'filename_prefix': ('STRING', {'default': 'ComfyUI'}), 'only_preview': ('BOOLEAN', {'default': False})}, 'hidden': {'prompt': 'PROMPT', 'extra_pnginfo': 'EXTRA_PNGINFO'}}
    RETURN_TYPES = ()
    FUNCTION = 'save'
    OUTPUT_NODE = True
    CATEGORY = 'EasyUse/Image'

    def save(self, images, filename_prefix='ComfyUI', only_preview=False, prompt=None, extra_pnginfo=None):
        if only_preview:
            PreviewImage().save_images(images, filename_prefix, prompt, extra_pnginfo)
            return ()
        else:
            return SaveImage().save_images(images, filename_prefix, prompt, extra_pnginfo)