python - 将 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:如果您不想就地修改，您可以只创建一个副本。