Python 和 PiCamera 颜色检测 bool 问题

Question

我对 Python 相当陌生，正在尝试编写一种 PiCamera 来拾取不同颜色并据此使用react的方法。我目前对每种颜色的尝试是拍照并检测所涉及的颜色。下面是一些示例代码。

def red_detect():
    # initialize the camera and grab a reference to the raw camera capture
    rawCapture = PiRGBArray(camera, size=(640, 480))

    # allow the camera to warmup
    time.sleep(0.1)

    # grab an image from the camera
    camera.capture(rawCapture, format="bgr")
    image = rawCapture.array

    # Convert BGR to HSV
    hsv = cv2.cvtColor(image, cv2.COLOR_BGR2HSV)

    lower_red = np.array([161, 155, 84],np.uint8)
    upper_red = np.array([179, 255, 255],np.uint8)
    red_mask = cv2.inRange(hsv, lower_red, upper_red)
    red = cv2.bitwise_and(image, image, mask=red_mask)

    return 255 in cv2.inRange(hsv, lower_red, upper_red)  

然后该代码将使用 if 语句进行评估，例如如果绿色== True:

但是，我的所有颜色都恢复为 True。我是不是拍错了照片？

Answer 1

如果你只是想检测图像的颜色，有更简单的方法:

from PIL import Image

image = Image.open('image.jpg')

w, h = image.size

pixel = []

for x in range(w):
    for y in range(h):
        r, g, b = image.getpixel((x, y))
        pixel.append([r, g, b])

avr = [sum(x)//len(x) for x in zip(*pixel)]

colors = dict(zip(['red','green','blue'],[i for i in avr]))
print(max(colors))

附注我在那里使用枕头模块。比我能提供的帮助，也是 python 新手