python - OpenCV python，如何以python方式将操作应用于单个 channel ？

Question

假设我有一个尺寸为 1000x1000x3 的 OpenCV 图像

假设我想应用每像素操作

if pixel == (255, 255, 255): 
     pixel = (0, 255, 0)

很明显，我可以只写 2 个 for 循环，然后那样做。然而，在 python 中访问像素的 for 循环非常慢。

有人告诉我有一种 pythonic 方法可以做这样的事情，例如 img = cv2.min(img, 255) 会有效地找到垫子像素值和 255 的最小值限制单个 channel 图像上的值范围。

但是，我不确定如何以类似的方式将每像素 if 条件应用于每个像素。

Answer 1

您可以使用 numpy 条件:

例如，将 5x5 矩阵中的偶数替换为 42:

import numpy as np

a = np.arange(5*5).reshape((5,5))
print(a)

输出:

array([[ 0,  1,  2,  3,  4],
   [ 5,  6,  7,  8,  9],
   [10, 11, 12, 13, 14],
   [15, 16, 17, 18, 19],
   [20, 21, 22, 23, 24]])

a[a%2==0] = 42
print(a)

这会产生:

array([[42,  1, 42,  3, 42],
   [ 5, 42,  7, 42,  9],
   [42, 11, 42, 13, 42],
   [15, 42, 17, 42, 19],
   [42, 21, 42, 23, 42]])

对于您的图像，您需要更多操作。它会是这样的:

myImg = np.arange(1000*1000*3).reshape((1000,1000,3)) # replace with your image
colorVecs = np.reshape(myImg,(-1,3)) #so that each array element is a 1x3 vector
colorVecs[colorVecs == [255,255,255]] = [0,255,0]  #perform your operation

filteredImg = np.reshape(colorVecs,(1000,1000,3)) #back to a normal image

有关 numpy bool 掩码的更多信息 here