CavinHuang-comfyui-nodes-docs/docs/SaltMaskCeilingRegion.md

Documentation

• Class name: SaltMaskCeilingRegion
• Category: SALT/Masking/Filter
• Output node: False
• Repo Ref: https://github.com/comfyanonymous/ComfyUI

SaltMaskCeilingRegion节点专门用于识别和提取给定遮罩集中的天花板区域。它对每个遮罩进行处理，以突出显示天花板区域，从而便于对这些特定区域进行集中分析或修改。这个节点在处理室内场景图像时特别有用，可以帮助isolate天花板区域以进行后续处理或特效应用。

Input types

Required

• masks
  • 这个参数是用于天花板区域识别的输入遮罩。它对于确定每个遮罩中对应于天花板区域的部分至关重要。输入的遮罩决定了节点的工作范围和最终输出的质量。
  • Comfy dtype: MASK
  • Python dtype: torch.Tensor

Output types

• MASKS
  • 输出的MASKS是经过处理后的遮罩，其中天花板区域已被识别并从原始输入遮罩中分离出来。这些处理后的遮罩可以用于进一步的图像处理任务，如天花板纹理替换、照明效果模拟等。
  • Comfy dtype: MASK
  • Python dtype: torch.Tensor

Usage tips

• Infra type: GPU
• Common nodes: unknown

Source code

class SaltMaskCeilingRegion:
    @classmethod
    def INPUT_TYPES(cls):
        return {
            "required": {
                "masks": ("MASK",),
            }
        }

    CATEGORY = f"{NAME}/Masking/Filter"

    RETURN_TYPES = ("MASK",)
    RETURN_NAMES = ("MASKS",)

    FUNCTION = "ceiling_region"
    
    def ceiling_region(self, masks):
        regions = []
        for mask in masks:
            pil_image = mask2pil(mask.unsqueeze(0))
            region_mask = MaskFilters.ceiling_region(pil_image)
            region_tensor = pil2mask(region_mask)
            regions.append(region_tensor)
        regions_tensor = torch.cat(regions, dim=0)
        return (regions_tensor,)