49 lines
1.2 KiB
Markdown
49 lines
1.2 KiB
Markdown
# 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
|
|
```python
|
|
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")
|
|
``` |