带有掩码和 RGB 图像的 Numpy/Scipy

Question

我正在尝试使用skikit learn 为RGB 图像创建蒙版。我想创建一个仅选择等于 [0,10,0] 的像素的蒙版，即绿色通道上的 10。然后只显示那些像素。这应该是直截了当的，类似于http://scikit-image.org/docs/dev/user_guide/numpy_images.html，但我很挣扎。

如果图像是加载的 jpg，我可以这样做

mask = image == [0,10,0]
image = image[mask]
viewer = ImageViewer(image)
viewer.show()

但是，我得到：

TypeError: Invalid dimensions for image data

如果我然后打印（掩码），我意识到不是每个像素的一系列真假，我得到：

[[ True False  True]
  [ True False  True]
  [ True False  True]
  ..., 
  [ True False  True]
  [ True False  True]
  [ True False  True]]

请注意，我图像中的第一个像素是黑色的。所以看起来它正在做的是将 [0,0,0] 与 [0,10,0] 进行比较，而不是引发 False，而是引发 True,False,True。

这似乎作为掩码失败了，因为我想每个像素都有 3 个掩码！

有没有人知道一种简单的方法来让这种遮罩与 RGB 图像一起使用？

谢谢

Answer 1

你可以得到 2D 掩码，沿最后一个轴全部减少 -

mask = (image == [0,10,0]).all(-1)

然后，image[mask] 将是仅包含 [0,10,0] 值的 (N,3) 形状数组，其中 N 是该特定 RGB 三元组的像素数。

因此，使用mask 显示蒙版图像或叠加层的步骤将取决于查看者。

---

对于图像中的原位编辑，这样我们就可以屏蔽掉不属于特定 RGB 三元组的所有内容，我们可以与蒙版相乘 -

image *= mask[...,None]

或者使用np.where创建一个带有选择机制的副本-

image_overlayed = np.where(mask[...,None], image, 0)

要获得3D 掩码（如果查看器需要此掩码），我们也可以沿通道复制掩码 -

np.repeat(mask[...,None],3,axis=2)