将2D numpy蒙版应用于灰度图像

Question

我有一个256×256像素的2D蒙版，其像素值为0或255，以及一个匹配的灰度图像。 两者都有形状（256，256）。

我想在灰度图像上应用蒙版，因此生成的图像仅包含原始灰度图像中的像素，其中相应的蒙版值为255（应保留原始灰度像素值）。

我无法使用

img_clean = img_grayscale[mask]

因为它为数组增加了新的维度（如numpy的文档所述）。

我该如何解决？

Answer 1

只需添加第三个选项并就地修改您的灰度图像：

img_grayscale[mask == 255] = 0

PS：如果您不想就地修改，则只能创建一个副本。

Answer 2

我认为这应该做的

np.where(mask == 255, img_grayscale, 0)

有关文档，请参见https://docs.scipy.org/doc/numpy/reference/generated/numpy.where.html 。

Answer 3

可以使用按位AND来将像素值设置为0 ，其中mask为0 ：

img_clean = img_grayscale & mask

由于掩码包含0或255 ，因此根据您的要求，值可以保持不变或设置为0 。