重庆理工大学学报(自然科学) ›› 2023, Vol. 37 ›› Issue (11): 64-72.

• 车辆工程 • 上一篇    下一篇

面向交通视频流的轻量化车辆检测模型

丁 华,刘 来   

  1. (江苏大学 汽车与交通工程学院,江苏 镇江 212013)
  • 出版日期:2023-12-14 发布日期:2023-12-14
  • 作者简介:丁华,男,博士,教授,主要从事智能驾驶研究,Email:dh@ujs.edu.cn;通信作者 刘来,男,硕士研究生,主要从 事车辆检测研究,Email:2319526290@qq.com。

Lightweight vehicle detection model for traffic video streams

  • Online:2023-12-14 Published:2023-12-14

摘要: 针对交通视频流中车辆检测模型内存占用大、部分特征难以提取等问题,提出了一 种改进的 YOLOv7网络模型。使用轻量化结构替换主干网络,引入的无参注意力机制在不增加 模型参数量的前提下提升特征提取能力,同时融合改进的双向特征金字塔网络提升检测精度。 为了增强模型的鲁棒性,对数据集中的样本进行了随机忽略区域的处理。将改进后的网络模型 在处理过的数据集上进行实验,结果表明参数量约为基础模型的 1/6,FPS提高到了 143.8, mAP0.5提升了 9.1%,验证了模型的优越性。

关键词: 交通视频流, 车辆检测, MobileNetV3, SimAM, BiFPN

中图分类号: 

  • U495