CavinHuang-comfyui-nodes-docs/docs/LayerMask%3A MaskPreview.md

Documentation

• Class name: MaskPreview
• Category: 😺dzNodes/LayerMask
• Output node: True
• Repo Ref: https://github.com/chflame163/ComfyUI_LayerStyle

用于生成Mask预览图像的节点。

Input types

Required

• mask
  • 输入的掩码。
  • Comfy dtype: MASK
  • Python dtype: torch.Tensor

Output types

• 无直接输出，通过保存预览图像来间接输出。

Usage tips

• Infra type: CPU

Source code

class MaskPreview(SaveImage):
    def __init__(self):
        self.output_dir = folder_paths.get_temp_directory()
        self.type = "temp"
        self.prefix_append = "_temp_" + ''.join(random.choice("abcdefghijklmnopqrstupvxyz1234567890") for x in range(5))
        self.compress_level = 4

    @classmethod
    def INPUT_TYPES(self):
        return {
            "required": {"mask": ("MASK",), },
        }

    FUNCTION = "mask_preview"
    CATEGORY = '😺dzNodes/LayerMask'
    OUTPUT_NODE = True

    def mask_preview(self, mask):
        if mask.dim() == 2:
            mask = torch.unsqueeze(mask, 0)
        preview = mask.reshape((-1, 1, mask.shape[-2], mask.shape[-1])).movedim(1, -1).expand(-1, -1, -1, 3)
        return self.save_images(preview, "MaskPreview")