Research and Implementation of Vehicle Target Detection Algorithm Based on Improved YOLOv5s
ZHOU Jinzhi
JING Ruiqi
WU Jing
LIU Mengyu
Abstract：Aiming at the requirements of the vehicle target detection algorithm in the actual traffic scene,such as occupying small resources,ensuring real-time performance and high accuracy,a vehicle target detection algorithm based on the improved YO-LOv5s is proposed.Firstly,GhostNet is introduced to improve the Backbone of YOLOv5s,which reduces the computation of the net-work and improves the detection speed.Secondly,the CBAM attention mechanism is integrated to improve the difficulty of accurate detection under various weather and light conditions.Then,Soft-NMS is used instead of NMS to reduce the problem of missing de-tection caused by traffic congestion.Finally,a comparative ablation experiment is conducted to verify the performance of the im-proved algorithm,and then it is deployed to the embedded device for testing.According to the experimental results,the resource oc-cupancy of the model is reduced by 34.76%under the condition that the improved algorithm guarantees high average accuracy,and the frame rate on the embedded platform can reach 29 frame/s,which can meet the requirements of practical applications.
Keywords：YOLOv5target detectionattention mechanismembedded platformTensorRT
Publication Date：2023-11-20
Online Publishing Date：2025-08-15(First online date of this platform, not the publication date of the document)
Pages：8( 2546-2552,2579 )