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

Documentation

• Class name: SaltMaskFloorRegion
• Category: SALT/Masking/Filter
• Output node: False

SaltMaskFloorRegion节点旨在识别和提取给定掩码中的地板区域。它会处理每个掩码以提取地板区域，并将其转换为表示地板区域的张量。

Input types

Required

• masks
  • 需要识别地板区域的输入掩码。这些掩码将被处理以提取地板区域，对节点的输出有重大影响。
  • Comfy dtype: MASK
  • Python dtype: torch.Tensor

Output types

• MASKS
  • 表示输入掩码中被隔离的地板区域的输出张量。
  • Comfy dtype: MASK
  • Python dtype: torch.Tensor

Usage tips

• Infra type: GPU
• Common nodes: unknown

Source code

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

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

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

    FUNCTION = "floor_region"

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