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

2.2 KiB

Documentation

该节点通过将图像保存在临时目录中来便利图像预览,允许用户可视化图像处理任务的输出,而不影响主要工作流程。

Input types

Required

  • enabled
    • 该参数控制是否激活图像预览功能,决定节点是否继续进行图像保存过程。
    • Comfy dtype: BOOLEAN
    • Python dtype: bool

Optional

  • images
    • 要预览和保存的输入图像;它们是节点操作的主要对象。
    • Comfy dtype: IMAGE
    • Python dtype: PIL.Image
  • prompt
    • 与图像相关联的可选文本描述,可用于为图像预览提供上下文或其他信息。
    • Comfy dtype: PROMPT
    • Python dtype: str
  • extra_pnginfo
    • 可以嵌入到保存的图像中的附加元数据,增强每个图像文件可用的信息。
    • Comfy dtype: EXTRA_PNGINFO
    • Python dtype: Dict[str, str]

Output types

  • images
    • 作为输入提供的原始图像,现在伴随着预览功能,允许在UI中进行可视化。
    • Comfy dtype: IMAGE
    • Python dtype: PIL.Image

Usage tips

  • Infra type: CPU

Source code

class SeargePreviewImage(nodes.SaveImage):

    def __init__(self):
        super().__init__()
        self.output_dir = folder_paths.get_temp_directory()
        self.type = 'temp'
        self.prefix_append = '_temp_' + ''.join((random.choice('abcdefghijklmnopqrstupvxyz') for _ in range(5)))

    @classmethod
    def INPUT_TYPES(cls):
        return {'required': {'enabled': ('BOOLEAN', {'default': True})}, 'optional': {'images': ('IMAGE',)}, 'hidden': {'prompt': 'PROMPT', 'extra_pnginfo': 'EXTRA_PNGINFO'}}
    RETURN_TYPES = ('IMAGE',)
    RETURN_NAMES = ('images',)
    FUNCTION = 'preview_images'
    CATEGORY = UI.CATEGORY_UI

    def preview_images(self, enabled, images=None, prompt=None, extra_pnginfo=None):
        if images is None or not enabled:
            return {'result': (images,), 'ui': {'images': list()}}
        saved_images = nodes.SaveImage.save_images(self, images, 'srg_sdxl_preview', prompt, extra_pnginfo)
        saved_images['result'] = (images,)
        return saved_images