我想更改 BGR 圖像元素。具體來說，如果第 2 個元素等于第 3 個元素，則它們都變為 0。arg1 = np.argwhere(img[:, :, 1] == img[:, :, 2])np.put(img[:, :, 1], arg1, 0)np.put(img[:, :, 2], arg1, 0)我嘗試過這個但不起作用。
1 回答
德瑪西亞99
TA貢獻1770條經驗 獲得超3個贊
您的代碼確實有效，但只是不是您期望的那樣。np.put期望多維矩陣的索引作為元組，同時np.argwhere給出行和列的二維數組。
為了使其變得更簡單，您可以使用布爾掩碼并完成工作 -
mask = img[:, : ,1] == img[:,:,2]
img[:, :, 0][mask] = 0
添加回答
舉報
0/150
提交
取消