Vehicle Target Detection Using the Improved YOLOv5s Algorithm

被引:1
|
作者
Dong, Zhaopeng [1 ]
机构
[1] Shanghai Ocean Univ, Coll Engn, Shanghai 201306, Peoples R China
来源
ELECTRONICS | 2024年 / 13卷 / 23期
关键词
YOLOv5s; vehicle inspection; DeepSORT; tracking and monitoring; model optimization;
D O I
10.3390/electronics13234672
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
This paper explores the application of the YOLOv5s algorithm integrated with the DeepSORT tracking detection algorithm in vehicle target detection, leveraging its advantages in data processing, loss function, network structure, and training strategy. Regarding detection frame regression, adopting Focal-EIOU can improve vehicle detection accuracy by precisely measuring overlap and better handle complex scenarios, enhancing the overall performance. The CoordConv convolution layer with more spatial position information is employed to enhance the original network structure's convolution layer and improve vehicle positioning accuracy. The principle and effectiveness of the Shuffle Attention mechanism are analyzed and added to the YOLOv5s network structure to enhance training, improve detection accuracy and running speed. And the DeepSORT tracking detection algorithm is designed to achieve high-speed operation and high-accuracy matching in target tracking, enabling efficient and reliable tracking of objects. Simultaneously, the network structure is optimized to enhance algorithmic speed and performance. To meet the requirements of vehicle detection in practical transportation systems, real-world vehicle images are collected as a dataset for model training to achieve accurate vehicle detection. The results show that the accuracy rate P of the improved YOLOv5s algorithm is increased by 0.484%, and mAP_0.5:0.95 reaches 92.221%, with an increase of 1.747%.
引用
收藏
页数:19
相关论文
共 50 条
  • [1] Pedestrian Target Detection in Subway Scene Using Improved YOLOv5s Algorithm
    Zhang Xiuzai
    Qiu Ye
    Zhang Chen
    LASER & OPTOELECTRONICS PROGRESS, 2023, 60 (06)
  • [2] Tank Armored Vehicle Target Detection Based on Improved YOLOv5s
    Li, Xinwei
    Mao, Yuxin
    Yu, Jianjun
    Mao, Zheng
    2024 4TH INTERNATIONAL CONFERENCE ON COMPUTER COMMUNICATION AND ARTIFICIAL INTELLIGENCE, CCAI 2024, 2024, : 1 - 6
  • [3] Lightweight ship target detection algorithm based on improved YOLOv5s
    Qian L.
    Zheng Y.
    Cao J.
    Ma Y.
    Zhang Y.
    Liu X.
    Journal of Real-Time Image Processing, 2024, 21 (01)
  • [4] Improved YOLOv5s UAV View Small Target Detection Algorithm
    Liu, Tao
    Gao, Yimeng
    Chai, Rui
    Li, Zhengtong
    Computer Engineering and Applications, 2024, 60 (01) : 110 - 121
  • [5] Improved YOLOv5s Small Target Smoke and Fire Detection Algorithm
    Wang, Yixu
    Xiao, Xiaoling
    Wang, Pengfei
    Xiang, Jiafu
    Computer Engineering and Applications, 2023, 59 (01) : 72 - 81
  • [6] UAV small target detection algorithm based on improved YOLOv5s
    Song, Yaolian
    Wang, Can
    Li, Dayan
    Liu, Xinyi
    Zhejiang Daxue Xuebao (Gongxue Ban)/Journal of Zhejiang University (Engineering Science), 2024, 58 (12): : 2417 - 2426
  • [7] SwinT-YOLOv5s: Improved YOLOv5s for Vehicle-mounted Infrared Target Detection
    Xin, Xiuli
    Pan, Feng
    Wang, Jiacheng
    Feng, Xiaoxue
    Shao, Liwei
    2022 41ST CHINESE CONTROL CONFERENCE (CCC), 2022, : 7326 - 7331
  • [8] Improved YOLOv5s Model for Vehicle Detection and Recognition
    Lu, Xingmin
    Song, Wei
    INTELLIGENT COMPUTING METHODOLOGIES, PT III, 2022, 13395 : 423 - 434
  • [9] Lightweight Vehicle Detection Based on Improved YOLOv5s
    Wang, Yuhai
    Xu, Shuobo
    Wang, Peng
    Li, Kefeng
    Song, Ze
    Zheng, Quanfeng
    Li, Yanshun
    He, Qiang
    SENSORS, 2024, 24 (04)
  • [10] Improved YOLOv5s for typical carp target detection
    Lyu, Chao
    Zhong, Wei-Chao
    Liu, Shuang
    AQUACULTURE INTERNATIONAL, 2024, 32 (04) : 4869 - 4884