python - 如何将所有黑色像素更改为白色(OpenCV)？

Question

我是 OpenCV 的新手，我不明白如何遍历和更改所有颜色代码为 RGB(0,0,0) 的黑色像素到白色 RGB( 255,255,255)。是否有任何函数或方法来检查所有像素，如果 RGB(0,0,0) 则使其变为 RGB(255,255,255)。

Answer 1

假设您的图像表示为形状为 (height, width, channels) 的 numpy 数组(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]