尝试仅显示所捕获图像的某些彩色像素时出错

Question

我正在尝试将捕获图像的所有像素（不是黄色）转换为白色。

我收到的错误消息是：

if jmg[i,j]!=[255,255,0]:
valueError: the truth value of an array with more than one element is ambiguous. use a.any() or a.all()

下面是我的代码：

import cv2
import picamera
import numpy
import time
from matplotlib import pyplot as plt
print("ready")
with picamera.PiCamera() as camera:
    camera.resolution=(400,400)
    time.sleep(1)
    camera.capture("/home/pi/rowdy.jpg")
    print("Done")
yellow=[255,255,0]
img=cv2.imread("/home/pi/rowdy.jpg",1)
jmg=cv2.cvtColor(img,cv2.COLOR_BGR2RGB)
for i in numpy.arange(399):
  for j in numpy.arange(399):
      if jmg[i,j]!=[255,255,0]:
          jmg[i,j]=[255,255,255]

plt.imshow(jmg)
plt.xticks([]),plt.yticks([])
plt.show()

Answer 1

jmg是一个由三个值组成的numpy数组。 当您将numpy数组与另一个数组（或列表）进行比较时，它将按element-wise进行比较。

if jmg[i,j]!=[255,255,0]: #np.array(255, 0, 0) != [255,255,0]
                          # -> np.array(False, True, False)

这意味着您需要执行错误消息中所说的操作，并使用any()或all()来确定所需的值。 在你的情况，你要检查，如果any这些值不匹配，这意味着你的元素智能逻辑将使这三个值中的一个True

if (jmg[i,j]!=[255,255,0]).any():
          jmg[i,j]=[255,255,255]

Answer 2

一种更快的方法是使用numpy库。 首先创建法师的副本，然后使用numpy.argwhere在所需位置替换像素值：

image1 = image.copy()
image1[np.argwhere(image != [255,255,0])] = [255,255,255]