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

Documentation

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

该节点对一组遮罩集合中的每个遮罩应用直方图均衡化，以增强遮罩内区域的对比度。它旨在通过调整强度分布来改善每个遮罩内特征的可见度和区分度。

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 SaltMaskHistogramEqualizationRegion:
    @classmethod
    def INPUT_TYPES(cls):
        return {
            "required": {
                "masks": ("MASK",),
            }
        }

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

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

    FUNCTION = "histogram_equalization"

    def histogram_equalization(self, masks):
        regions = []
        for mask in masks:
            pil_image = mask2pil(mask.unsqueeze(0))
            image_array = np.array(pil_image.convert('L'))            
            equalized = cv2.equalizeHist(image_array)
            equalized_pil = ImageOps.invert(Image.fromarray(equalized))
            region_tensor = pil2mask(equalized_pil)
            regions.append(region_tensor)

        regions_tensor = torch.cat(regions, dim=0)
        return (regions_tensor,)