使用切片用黑色替换特定颜色？

Question

第一次在stackoverflow上提问！ 我正在尝试使用切片替换特定的像素颜色。我有兴趣用黑色代替粉红色。

 colors = [(0, 0, 0), (255, 0, 255)]
 img = cv2.imread('Untitled.png')  # Random image containing some pink pixels
 pink = img[:, :, :] == np.array(colors[1])  # Boolean array with TRUE @ all pink indices

当我尝试使用此功能进行替换时

img[pink, :] = np.array(colors[0])  # Replace with black

我收到以下错误

img[pink, :] = np.array(colors[0])
IndexError: too many indices for array

img 和粉色是相同的尺寸和大小。我做错了什么？

Answer 1

这应该适合你：

import cv2

image = cv2.imread('test.png')
image[np.where((image==[255,0,255]).all(axis=2))] = [0,0,0]
cv2.imwrite('test_out.png', image)