做一个opencv、Python的摄像头检测行人并框出行人
时间: 2024-01-25 18:04:46 浏览: 170
yolov5应用,只需要opencv库即可,包括c++和python版本。支持对图片和视频进行检测和摄像头实时检测.zip
好的,这个问题可以分成以下几个步骤来实现:
1. 导入OpenCV和NumPy库:
```python
import cv2
import numpy as np
```
2. 加载Haar级联分类器文件:
```python
pedestrian_cascade = cv2.CascadeClassifier('haarcascade_fullbody.xml')
```
这里使用的是已经训练好的Haar级联分类器文件,可以在OpenCV官网上下载。
3. 初始化摄像头并进行行人检测:
```python
cap = cv2.VideoCapture(0)
while True:
# 读取摄像头中的图像
ret, frame = cap.read()
# 转换为灰度图像
gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
# 检测行人
pedestrians = pedestrian_cascade.detectMultiScale(gray, 1.3, 5)
# 绘制矩形框
for (x, y, w, h) in pedestrians:
cv2.rectangle(frame, (x, y), (x+w, y+h), (0, 255, 0), 2)
# 显示图像
cv2.imshow('Pedestrian Detection', frame)
# 按下q键退出
if cv2.waitKey(1) == ord('q'):
break
# 释放摄像头资源
cap.release()
# 关闭所有窗口
cv2.destroyAllWindows()
```
在上述代码中,我们首先使用`cv2.VideoCapture()`函数初始化摄像头,并不断读取摄像头中的图像。然后将读取到的图像转换为灰度图像,再使用`pedestrian_cascade.detectMultiScale()`函数进行行人检测,返回的是检测到的行人的矩形框坐标。最后,在原图像上绘制矩形框,使用`cv2.imshow()`函数显示图像,按下“q”键退出程序。
注意:如果没有检测到行人,返回的`pedestrians`为空,需要对这种情况进行处理,否则会报错。
阅读全文