节点文献
基于YOLOv5的目标识别追踪模型轻量化
Lightweight Object Recognition Rracking Model Based on YOLOv5
【摘要】 道路目标检测环节是自动驾驶领域的关键技术之一,随着人工智能的发展应用逐渐广泛。文章基于YOLOv5网络提出一种新的目标检测方法,改进包括融合了ShuffleNet V2中的模块,使用GhostConv改造了传统的Conv模块等。先在不同道路环境中实时采集视频流,并进行图片和视频流的标注。在主干网络中融入ShuffleNet V2中的模块并使用GhostConv模块改进Conv模块,在降低模型权重的同时对目标检测精度影响较小。将标注完成后的图片输入改进后的YOLOv5网络进行训练,并将得到后的模型与Deep SORT算法结合,进行目标检测追踪。实验结果表明,所得结果权重大小下降许多,而目标检测精确度有所上升。改进后的网络更加轻便,易于部署在边缘嵌入式设备上。
【Abstract】 The road target detection is one of the key technologies in the field of autonomous driving.With the development of artificial intelligence, the application of road target detection is gradually widespread. This paper proposes a new target detection method based on the YOLOv5 network. The improvements include the integration of modules in ShuffleNet V2 and the use of GhostConv to transform the traditional Conv module. First, real-time video streams are collected in different road environments, and pictures and video streams are annotated. The modules in ShuffleNet V2 are integrated into the backbone network and the GhostConv module is used to improve the Conv module,which reduces the weight of the model and has less impact on the target detection accuracy. The marked images are input to the improved YOLOv5 network for training, and the obtained model is combined with the Deep SORT algorithm for target detection and tracking. The experimental results show that the weight of the obtained results decreases a lot, while the target detection accuracy increases. The improved network is lighter and easier to deploy on edge embedded devices.
【Key words】 Road target detection; YOLOv5; Model compression; Target tracking;
- 【文献出处】 汽车实用技术 ,Automobile Applied Technology , 编辑部邮箱 ,2023年05期
- 【分类号】U463.6;TP391.41
- 【下载频次】491