写一段摄像头色块识别程序python
时间: 2023-08-15 13:03:15 浏览: 103
用Python获取摄像头并实时控制人脸的实现示例
好的,下面是一个使用摄像头实时识别特定颜色色块的 Python 程序:
```python
import cv2
import numpy as np
# 定义要识别的颜色范围
lower_color = np.array([25, 50, 50])
upper_color = np.array([35, 255, 255])
# 初始化摄像头
cap = cv2.VideoCapture(0)
while True:
# 获取当前帧
ret, frame = cap.read()
# 转换颜色空间
hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
# 根据颜色范围进行二值化
mask = cv2.inRange(hsv, lower_color, upper_color)
# 寻找轮廓
contours, hierarchy = cv2.findContours(mask, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
# 遍历轮廓
for contour in contours:
# 计算轮廓面积
area = cv2.contourArea(contour)
# 忽略面积较小的轮廓
if area < 100:
continue
# 计算轮廓的外接矩形
x, y, w, h = cv2.boundingRect(contour)
# 在原图上绘制外接矩形
cv2.rectangle(frame, (x, y), (x + w, y + h), (0, 255, 0), 2)
# 显示图像
cv2.imshow('frame', frame)
# 按下 q 键退出程序
if cv2.waitKey(1) & 0xFF == ord('q'):
break
# 释放摄像头并关闭窗口
cap.release()
cv2.destroyAllWindows()
```
这个程序将打开计算机上的默认摄像头,并实时显示摄像头捕捉到的图像。程序将根据定义的颜色范围对图像进行二值化处理,并寻找二值化图像中的轮廓。程序将忽略面积较小的轮廓,并在原图上绘制外接矩形。在程序运行时,你可以将一个颜色块放在摄像头前面,程序将识别并绘制出颜色块的外接矩形。按下 `q` 键即可退出程序。
阅读全文