Python OpenCV将所有黑色像素更改为白色

我是 OpenCV 的新手，我不明白如何遍历和更改所有颜色代码RGB(0,0,0)为白色的黑色像素RGB(255,255,255)。是否有任何功能或方法来检查所有像素以及是否RGB(0,0,0)使其成为RGB(255,255,255).

假设您的图像表示为一个numpy形状数组(height, width, channels)（cv2.imread返回什么），您可以执行以下操作：

height, width, _ = img.shape

for i in range(height):
    for j in range(width):
        # img[i,j] is the RGB pixel at position (i, j)
        # check if it's [0, 0, 0] and replace with [255, 255, 255] if so
        if img[i,j].sum() == 0:
            img[i, j] = [255, 255, 255]

一种更快的基于掩码的方法如下所示：

# get (i, j) positions of all RGB pixels that are black (i.e. [0, 0, 0])
black_pixels = np.where(
    (img[:, :, 0] == 0) & 
    (img[:, :, 1] == 0) & 
    (img[:, :, 2] == 0)
)

# set those pixels to white
img[black_pixels] = [255, 255, 255]