用opencv识别颜色,按下c键识别绿色,再次按下c键黄色,再次按下c键识别灰色
时间: 2024-05-06 09:21:40 浏览: 95
基于opencv的颜色识别
好的,谢谢您的提问。以下是我的回答:
您可以使用OpenCV中的cv2.inRange()函数来识别颜色。首先,您需要读取视频流或图像,然后转换为HSV颜色空间。在HSV颜色空间中,您可以轻松地识别特定颜色的像素。接着,您可以使用cv2.inRange()函数来将图像转换为二进制图像,其中仅包含在指定范围内的像素值。最后,您可以使用cv2.bitwise_and()函数将原始图像和二进制图像合并,然后显示结果。
对于您的要求,下面是一个基于OpenCV的示例代码,可以识别绿色、黄色和灰色:
```
import cv2
import numpy as np
cap = cv2.VideoCapture(0)
while True:
ret, frame = cap.read()
hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
# 绿色
lower_green = np.array([45, 100, 50])
upper_green = np.array([75, 255, 255])
# 黄色
lower_yellow = np.array([20, 100, 100])
upper_yellow = np.array([40, 255, 255])
# 灰色
lower_gray = np.array([0, 0, 70])
upper_gray = np.array([180, 30, 255])
if cv2.waitKey(1) & 0xFF == ord('c'):
mask_green = cv2.inRange(hsv, lower_green, upper_green)
mask_yellow = cv2.inRange(hsv, lower_yellow, upper_yellow)
mask_gray = cv2.inRange(hsv, lower_gray, upper_gray)
if np.sum(mask_green) > 0:
print('Green color detected')
elif np.sum(mask_yellow) > 0:
print('Yellow color detected')
elif np.sum(mask_gray) > 0:
print('Gray color detected')
cv2.imshow('frame', frame)
if cv2.waitKey(1) & 0xFF == ord('q'):
break
cap.release()
cv2.destroyAllWindows()
```
在这个示例中,我们首先读取视频流,然后将每一帧转换为HSV颜色空间。然后,我们定义了三个颜色范围(绿色、黄色和灰色),并使用cv2.inRange()函数将每个颜色转换为相应的二进制图像。如果用户按下'c'键,我们检查哪个颜色具有最高的像素值,然后打印相应的消息。最后,我们将原始帧显示出来,并等待用户按下'q'键退出程序。
希望这个示例能够帮到您。如果您还有其他的问题或需要进一步的帮助,请随时问我。
阅读全文