编写一段摄像头色块识别程序
时间: 2023-12-09 11:05:37 浏览: 175
以下是基于Python和OpenCV的摄像头色块识别程序:
```
import cv2
import numpy as np
cap = cv2.VideoCapture(0)
while True:
# Capture frame-by-frame
ret, frame = cap.read()
# Convert BGR to HSV
hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
# Define range of blue color in HSV
lower_blue = np.array([110,50,50])
upper_blue = np.array([130,255,255])
# Threshold the HSV image to get only blue colors
mask = cv2.inRange(hsv, lower_blue, upper_blue)
# Bitwise-AND mask and original image
res = cv2.bitwise_and(frame,frame, mask= mask)
# Find contours
contours, hierarchy = cv2.findContours(mask, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
# Draw contours
cv2.drawContours(frame, contours, -1, (0,255,0), 3)
# Show result
cv2.imshow('frame',frame)
cv2.imshow('mask',mask)
cv2.imshow('res',res)
# Exit on ESC
k = cv2.waitKey(5) & 0xFF
if k == 27:
break
# When everything done, release the capture
cap.release()
cv2.destroyAllWindows()
```
这个程序使用OpenCV的`cv2.VideoCapture`函数从默认摄像头捕获帧,并将每个帧转换为HSV颜色空间。然后,程序使用`cv2.inRange`函数通过指定颜色的HSV范围来创建掩膜,并使用`cv2.findContours`函数查找掩膜中的轮廓,使用`cv2.drawContours`函数绘制轮廓线,并使用`cv2.imshow`函数显示结果图像。
注意,这个程序仅仅是一个简单的示例,可能需要根据你的具体应用场景进行修改。
阅读全文