如何堆叠多个二进制掩码以创建用于多类分割的单个掩码？

Question

所以我有 5 个用于分割的类（不包括背景），所有掩码都是不同文件夹中的二进制掩码。如何组合二进制掩码为所有类创建单个掩码以进行多类分割？

我尝试在不同颜色通道中使用 2 个类，但由于只有 3 个颜色通道，我只能将背景和 2 个类合并到一个蒙版中。

python 代码：

new_img = np.zeros((height, width,3))

new_img[:,:,0] = new_img[:,:,0] + class1[:,:,0]

new_img[:,:,1] = new_img[:,:,1] + class2[:,:,0]

还有其他方法可以做到这一点吗？ 我可以使用二进制掩码进行多类分割而不合并它们吗？

谢谢

Answer 1

您可以通过制作具有 6 个通道的蒙版来组合蒙版：

mask = np.stack([mask0, mask1, mask2, mask3, mask4, mask5], axis=-1)

假设0..5 中i 的每个maski 是一个二维数组。

如果要可视化图像，则必须将每个类映射到一种颜色。例如：

@np.vectorize(signature='(6)->(3)')
def mapcolors(x):
    i = np.argmax(x)
    if i == 0:
        return [0.,0.,0.]
    elif i == 1:
        return [0.,0.,1.]
    elif i == 2:
        return [0.,1.,0.]
    elif i == 3:
        return [1.,0.,0.]
    elif i == 4:
        return [0.,1.,1.]
    else:
        return [1.,1.,0.]

image = mapcolors(mask)