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 条
  • [41] An Improved YOLOv5s Algorithm for Object Detection with an Attention Mechanism
    Jiang, Tingyao
    Li, Cheng
    Yang, Ming
    Wang, Zilong
    ELECTRONICS, 2022, 11 (16)
  • [42] Vehicle detection based on improved YOLOv5s using coordinate attention and decoupled head
    Shen, Xuanjing
    Liu, Tongzhuang
    Wang, Yu
    JOURNAL OF ELECTRONIC IMAGING, 2023, 32 (06)
  • [43] Improved small target detection algorithm based on multiattention and YOLOv5s for traffic sign recognition
    Ma, Ge
    Li, Hongwei
    Yan, Ziwei
    Liu, Zhijie
    Zhao, Zhijia
    Gongcheng Kexue Xuebao/Chinese Journal of Engineering, 2024, 46 (09): : 1647 - 1658
  • [44] YOLOv5s-DSD: An Improved Aerial Image Detection Algorithm Based on YOLOv5s
    Sun, Chaoyue
    Chen, Yajun
    Xiao, Ci
    You, Longxiang
    Li, Rongzhen
    SENSORS, 2023, 23 (15)
  • [45] YOLOv5s-GTB: An Improved Underwater Garbage Detection Algorithm Based on YOLOv5s
    Liu, Yijie
    Wang, Haichen
    2024 9TH INTERNATIONAL CONFERENCE ON COMPUTER AND COMMUNICATION SYSTEMS, ICCCS 2024, 2024, : 33 - 38
  • [46] Improved Method for Apple Fruit Target Detection Based on YOLOv5s
    Wang, Huaiwen
    Feng, Jianguo
    Yin, Honghuan
    AGRICULTURE-BASEL, 2023, 13 (11):
  • [47] Lightweight Target Detection for Coal and Gangue Based on Improved Yolov5s
    Cao, Zhenguan
    Fang, Liao
    Li, Zhuoqin
    Li, Jinbiao
    PROCESSES, 2023, 11 (04)
  • [48] Number detection of cylindrical objects based on improved Yolov5s algorithm
    Wang, Yachao
    Liu, Yaju
    Wang, Dongxuan
    Liu, Yu
    INTELLIGENT DECISION TECHNOLOGIES-NETHERLANDS, 2024, 18 (01): : 441 - 456
  • [49] Application of improved YOLOv5s algorithm in traffic sign detection and recognition
    Guo, Junbin
    Yu, Lin
    Yu, Chuanqiang
    Guofang Keji Daxue Xuebao/Journal of National University of Defense Technology, 2024, 46 (06): : 123 - 130
  • [50] Strip Pickling Defect Detection Based on Improved YOLOv5s Algorithm
    Zhu, Xiaoyan
    Zhang, Mingyu
    Liu, Yong
    Wan, Xin
    Wang, Xiwen
    Zhou, Peng
    2023 35TH CHINESE CONTROL AND DECISION CONFERENCE, CCDC, 2023, : 2234 - 2238