將2D numpy蒙版應用於灰度圖像

Question

我有一個像素值爲0或255的256×256像素2D蒙版，以及匹配的灰度圖像。兩者都有形狀（256,256）。

我想在灰度圖像上應用蒙版，因此生成的圖像只包含原始灰度圖像中的像素，其中相應的蒙版值爲255（應保留原始灰度像素值）。

我不使用

img_clean = img_grayscale[mask] 

，因爲它增加了一個新的層面的陣列（如numpy的的文檔中提到）能夠做到這一點。

我該如何解決這個問題？

Answer 1

我想這應該做

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

見https://docs.scipy.org/doc/numpy/reference/generated/numpy.where.html的文檔工作。

Answer 2

甲bitwise AND可用於以像素值設置爲0其中所述掩模是0：

img_clean = img_grayscale & mask 

作爲掩模包含任一0或255值要麼保持相同或設置爲0根據您的要求。

Answer 3

我想補充一個選項，第三和修改灰度圖像就地：

img_grayscale[mask == 255] = 0 

PS：你可以創建一個副本，如果你不想就地修改。