1

我对 Python 还很陌生，并试图编写一种 PiCamera 来获取不同颜色并据此做出反应的方式。我目前对每种颜色的尝试是拍照并检测所涉及的颜色。下面是一些示例代码。

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 语句评估该代码，例如 if green == True:

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

4

1 回答 1

0

如果您只想检测图像的颜色，有更简单的方法：

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))

PS我在那里使用枕头模块。比我能帮上忙，也是 python 新手

于 2019-11-23T07:05:38.983 回答