Color To Mask

节点功能:识别输入图像中与指定RGB颜色值相近的像素,并将这些像素转换为掩码。这个节点可以处理单张图像或批量图像,支持阈值调整和掩码反转。

输入参数

参数名称说明
images输入的 RGB 图像批次。

输出参数

参数名称说明
MASK输出的单通道二值掩码图像。

控件参数

参数名称说明
invert是否反转输出的掩码。
red要匹配的红色通道值(0-255)。
green要匹配的绿色通道值(0-255)。
blue要匹配的蓝色通道值(0-255)。
threshold距离阈值,允许的颜色偏差范围(单位为欧氏距离)。
per_batch每批次处理的图像数量,避免显存溢出。

如下图所示我们上传原图,通过设置对应的颜色区域(255,0,0)表示红色,然后设置相应的threshold就可以识别到原图中颜色对应的范围进行蒙版的输出。


threshold: 整数值,默认为10,范围0-255,控制颜色匹配的容差。值越大,容差越大。但是过大的话,会把不是设定颜色的颜色也分割出来。