python - OpenCV:将特定 BGR 值的所有像素设置为另一个 BGR 值

Question

我正在将 OpenCV 与 Python 结合使用。我有一张图片，我想要做的是将 BGR 值 [0, 0, 255] 的所有像素设置为 [0, 255, 255]。

我问了一个previous question关于如何对图像进行后处理，从答案中我了解到使用索引数组进行索引，例如: 图片[图片 > 128] = 255

我理解这是如何工作的，因为 image > 128 将返回一个满足条件的多维索引数组，然后我将这个数组应用于图像并将它们设置为 255。但是，我得到对如何将其扩展为为数组赋值感到困惑。

我尝试执行以下操作:

      red = np.array([0, 0, 255])
      redIndex = np.where(np.equal(image, red))
      image[redIndex] = np.array([0, 255, 255])

但它不起作用，出现错误:

ValueError: array is not broadcastable to correct shape

有没有有效的方法来处理这个问题？

Answer 1

考虑一个像数组这样的图像:

>>> red
array([[[  0,   0, 255],
        [  0,   0, 255],
        [  0,   0, 255],
        [  0,   0, 255],
        [  0,   0, 255]],

       [[  0,   0, 255],
        [  0,   0, 255],
        [  0,   0, 255],
        [  0,   0, 255],
        [  0,   0, 255]]])

它的所有元素都是[0,0,255]。它的形状是 2x5x3。试想其中还有其他值(value)。 (我无法创建所有这些)。

现在您找到 [0,0,255] 所在的位置并将它们更改为 [0,255,255]。您可以按如下方式进行:

>>> red[np.where((red == [0,0,255]).all(axis = 2))] = [0,255,255]

现在检查红色。

>>> red
array([[[  0, 255, 255],
        [  0, 255, 255],
        [  0, 255, 255],
        [  0, 255, 255],
        [  0, 255, 255]],

       [[  0, 255, 255],
        [  0, 255, 255],
        [  0, 255, 255],
        [  0, 255, 255],
        [  0, 255, 255]]])

希望这就是你想要的。

测试结果:

从此链接查看代码:https://stackoverflow.com/a/11072667/1134940

我想按照问题的要求将所有红色像素更改为黄色。

所以我在下面添加了这段代码:

im2[np.where((im2 == [0,0,255]).all(axis = 2))] = [0,255,255]

下面是我得到的结果:

如果我想把绿色地面变成黄色地面怎么办:

im2[np.where((im2 == [0,255,0]).all(axis = 2))] = [0,255,255]

结果: