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