当前位置：首页 > news >正文

实战 | 基于YOLOv10的车辆追踪与测速实战【附源码+步骤详解】

news 文章来源：https://blog.csdn.net/qq_42589613/article/details/139710732 2024/10/31 23:20:35

《博主简介》

小伙伴们好，我是阿旭。专注于人工智能、AIGC、python、计算机视觉相关分享研究。
✌更多学习资源，可关注公-仲-hao:【阿旭算法与机器学习】，共同学习交流~
👍感谢小伙伴们点赞、关注！

《------往期经典推荐------》

一、AI应用软件开发实战专栏【链接】

项目名称	项目名称
1.【人脸识别与管理系统开发】	2.【车牌识别与自动收费管理系统开发】
3.【手势识别系统开发】	4.【人脸面部活体检测系统开发】
5.【图片风格快速迁移软件开发】	6.【人脸表表情识别系统】
7.【YOLOv8多目标识别与自动标注软件开发】	8.【基于YOLOv8深度学习的行人跌倒检测系统】
9.【基于YOLOv8深度学习的PCB板缺陷检测系统】	10.【基于YOLOv8深度学习的生活垃圾分类目标检测系统】
11.【基于YOLOv8深度学习的安全帽目标检测系统】	12.【基于YOLOv8深度学习的120种犬类检测与识别系统】
13.【基于YOLOv8深度学习的路面坑洞检测系统】	14.【基于YOLOv8深度学习的火焰烟雾检测系统】
15.【基于YOLOv8深度学习的钢材表面缺陷检测系统】	16.【基于YOLOv8深度学习的舰船目标分类检测系统】
17.【基于YOLOv8深度学习的西红柿成熟度检测系统】	18.【基于YOLOv8深度学习的血细胞检测与计数系统】
19.【基于YOLOv8深度学习的吸烟/抽烟行为检测系统】	20.【基于YOLOv8深度学习的水稻害虫检测与识别系统】
21.【基于YOLOv8深度学习的高精度车辆行人检测与计数系统】	22.【基于YOLOv8深度学习的路面标志线检测与识别系统】
23.【基于YOLOv8深度学习的智能小麦害虫检测识别系统】	24.【基于YOLOv8深度学习的智能玉米害虫检测识别系统】
25.【基于YOLOv8深度学习的200种鸟类智能检测与识别系统】	26.【基于YOLOv8深度学习的45种交通标志智能检测与识别系统】
27.【基于YOLOv8深度学习的人脸面部表情识别系统】	28.【基于YOLOv8深度学习的苹果叶片病害智能诊断系统】
29.【基于YOLOv8深度学习的智能肺炎诊断系统】	30.【基于YOLOv8深度学习的葡萄簇目标检测系统】
31.【基于YOLOv8深度学习的100种中草药智能识别系统】	32.【基于YOLOv8深度学习的102种花卉智能识别系统】
33.【基于YOLOv8深度学习的100种蝴蝶智能识别系统】	34.【基于YOLOv8深度学习的水稻叶片病害智能诊断系统】
35.【基于YOLOv8与ByteTrack的车辆行人多目标检测与追踪系统】	36.【基于YOLOv8深度学习的智能草莓病害检测与分割系统】
37.【基于YOLOv8深度学习的复杂场景下船舶目标检测系统】	38.【基于YOLOv8深度学习的农作物幼苗与杂草检测系统】
39.【基于YOLOv8深度学习的智能道路裂缝检测与分析系统】	40.【基于YOLOv8深度学习的葡萄病害智能诊断与防治系统】
41.【基于YOLOv8深度学习的遥感地理空间物体检测系统】	42.【基于YOLOv8深度学习的无人机视角地面物体检测系统】
43.【基于YOLOv8深度学习的木薯病害智能诊断与防治系统】	44.【基于YOLOv8深度学习的野外火焰烟雾检测系统】
45.【基于YOLOv8深度学习的脑肿瘤智能检测系统】	46.【基于YOLOv8深度学习的玉米叶片病害智能诊断与防治系统】
47.【基于YOLOv8深度学习的橙子病害智能诊断与防治系统】	48.【车辆检测追踪与流量计数系统】
49.【行人检测追踪与双向流量计数系统】	50.【基于YOLOv8深度学习的反光衣检测与预警系统】
51.【危险区域人员闯入检测与报警系统】	52.【高密度人脸智能检测与统计系统】

二、机器学习实战专栏【链接】，已更新31期，欢迎关注，持续更新中~~
三、深度学习【Pytorch】专栏【链接】
四、【Stable Diffusion绘画系列】专栏【链接】
五、YOLOv8改进专栏【链接】，持续更新中~~
六、YOLO性能对比专栏【链接】，持续更新中~

《------正文------》

引言

在人工智能神经网络和计算机视觉的世界里，识别并跟踪物体是一项至关重要的技术。本文主要介绍一个深度学习项目，该项目利用YOLOv10算法来检测车辆并且测量马路上汽车的速度。通过这个项目，您可以更深入地理解这些算法是如何融入我们的日常解决方案中的。

实现效果

在这里插入图片描述

实现步骤

下载yolov10源码

下载地址：https://github.com/THU-MIG/yolov10

在这里插入图片描述

安装运行环境

conda create -n yolov10 python=3.9
conda activate yolov10
pip install -r requirements.txt
pip install -e .

下载v10n模型

下载官方提供的与训练模型，这里我们直接使用yolov10n.pt。
在这里插入图片描述

代码测试

运行完整代码：

from ultralytics import YOLOv10
from ultralytics.solutions import speed_estimation
import cv2# 加载YOLOv10模型
model = YOLOv10("yolov10n.pt")
# 获取模型中的对象名称
names = model.model.names# 打开视频文件
cap = cv2.VideoCapture("9.mp4")# 获取视频的宽度、高度和帧率
w, h, fps = (int(cap.get(x)) for x in (cv2.CAP_PROP_FRAME_WIDTH, cv2.CAP_PROP_FRAME_HEIGHT, cv2.CAP_PROP_FPS))# 创建视频写入器，用于输出处理后的视频
video_writer = cv2.VideoWriter("out.avi",cv2.VideoWriter_fourcc(*'mp4v'),fps,(w, h))# 设置测速线段的两个端点
line_pts = [(0, 180), (640, 180)]# 初始化速度估计器
speed_obj = speed_estimation.SpeedEstimator()
# 设置速度估计器的参数，包括测速线段、对象名称和是否显示图像
speed_obj.set_args(reg_pts=line_pts,names=names,view_img=True)# 循环读取视频帧
while cap.isOpened():# 读取一帧success, im0 = cap.read()# 如果读取失败，则退出循环if not success:breaktracks = model.track(im0, persist=True, show=False)im0 = speed_obj.estimate_speed(im0, tracks)video_writer.write(im0)# 释放视频读取器和写入器
cap.release()
video_writer.release()
# 销毁所有OpenCV窗口
cv2.destroyAllWindows()

代码详细解读

下面是对上述代码的详细解释说明：

导入必要的库：
```
from ultralytics import YOLOv10
from ultralytics.solutions import speed_estimation
import cv2
```
这段代码导入了ultralytics库中的YOLOv10模型和速度估计模块，以及OpenCV库，用于视频处理。
加载YOLOv10模型：
```
model = YOLOv10("yolov10n.pt")
names = model.model.names
```
这里加载了预训练的YOLOv10模型，用于识别车辆。

读取视频文件：

cap = cv2.VideoCapture("9.mp4")
assert cap.isOpened(), "Error reading video file"
w, h, fps = (int(cap.get(x)) for x in (cv2.CAP_PROP_FRAME_WIDTH, cv2.CAP_PROP_FRAME_HEIGHT, cv2.CAP_PROP_FPS))

这段代码读取名为"9.mp4"的视频文件，并获取视频的宽度、高度和帧率。

创建视频写入器：
```
video_writer = cv2.VideoWriter("out.avi", cv2.VideoWriter_fourcc(*'mp4v'), fps, (w, h))
```
这里创建了一个视频写入器，用于将处理后的视频帧写入新的视频文件。
设置速度估计参数：
```
line_pts = [(0, 180), (640, 180)]
speed_obj = speed_estimation.SpeedEstimator()
speed_obj.set_args(reg_pts=line_pts, names=names, view_img=True)
```
这段代码设置了用于速度估计的参考线（line_pts），并初始化了速度估计器，同时设置了相关参数。
速度计算的基本原理：
SpeedEstimator函数通过存储随时间推移的跟踪位置来处理帧，因此通过比较当前位置与定义区域内的先前位置来计算每个检测到的物体的速度，从而可以通过物体在该区域内移动所需的时间来估算物体的速度，这遵循了物理学中一个非常著名的术语：
v = Δs/Δt
其中，v为速度，Δs为位移（距离），Δt为时间间隔。

视频帧处理循环：

while cap.isOpened():success, im0 = cap.read()if not success:print("Video frame is empty or video processing has been successfully completed.")breaktracks = model.track(im0, persist=True, show=False)im0 = speed_obj.estimate_speed(im0, tracks)video_writer.write(im0)

在这个循环中，代码逐帧读取视频，使用YOLOv10模型进行对象跟踪，然后使用速度估计器估计对象的移动速度，并将结果写入新的视频文件。

释放资源：
```
cap.release()
video_writer.release()
cv2.destroyAllWindows()
```
最后，释放视频读取器和写入器的资源，并关闭所有OpenCV创建的窗口。

总结

这段代码的主要功能是使用YOLOv10模型对视频中的对象进行跟踪，并估计它们通过特定线段的速度。但是该方法受硬件检测速度等影响，计算出的速度不是特别准确。如果需要准确的计算车速，最好结合实际场景给出距离参考对象，这样计算出的速度才会更加准确。

好了，这篇文章就介绍到这里，感谢点赞关注，更多精彩内容持续更新中~

关注文末名片G-Z-H：【阿旭算法与机器学习】，可获取更多干货学习资源

引言