Exploration of Vehicle Target Detection Method Based on Lightweight YOLOv5 Fusion Background Modeling

被引:6
作者
Zhao, Qian [1 ]
Ma, Wenyue [1 ]
Zheng, Chao [2 ]
Li, Lu [2 ]
机构
[1] Xian Univ Sci & Technol, Sch Commun & Informat Engn, Xian 710054, Peoples R China
[2] Xian Key Lab Network Convergence Commun, Xian 710054, Peoples R China
来源
APPLIED SCIENCES-BASEL | 2023年 / 13卷 / 07期
基金
中国国家自然科学基金;
关键词
YOLOv5; target detection; ViBe; Ghostnet; CA mechanism; NETWORKS;
D O I
10.3390/app13074088
中图分类号
O6 [化学];
学科分类号
0703 ;
摘要
Due to the explosive increase per capita in vehicle ownership in China brought about by the continuous development of the economy and society, many negative impacts have arisen, making it necessary to establish the smart city system that has rapidly developing vehicle detection technology as its data acquisition system. This paper proposes a lightweight detection model based on an improved version of YOLOv5 to address the problem of missed and false detections caused by occlusion during rush hour vehicle detection in surveillance videos. The proposed model replaces the BottleneckCSP structure with the Ghostnet structure and prunes the network model to speed up inference. Additionally, the Coordinate Attention Mechanism is introduced to enhance the network's feature extraction and improve its detection and recognition ability. Distance-IoU Non-Maximum Suppression replaces Non-Maximum Suppression to address the issue of false detection and omission when detecting congested targets. Lastly, the combination of the five-frame differential method with VIBE and MD-SILBP operators is used to enhance the model's feature extraction capabilities for vehicle contours. The experimental results show that the proposed model outperforms the original model in terms of the number of parameters, inference ability, and accuracy when applied to both the expanded UA-DETRAC and a self-built dataset. Thus, this method has significant industrial value in intelligent traffic systems and can effectively improve vehicle detection indicators in traffic monitoring scenarios.
引用
收藏
页数:20
相关论文
共 26 条
[11]   Scale-Aware Fast R-CNN for Pedestrian Detection [J].
Li, Jianan ;
Liang, Xiaodan ;
Shen, Shengmei ;
Xu, Tingfa ;
Feng, Jiashi ;
Yan, Shuicheng .
IEEE TRANSACTIONS ON MULTIMEDIA, 2018, 20 (04) :985-996
[12]  
Li yongshang, 2022, Computer Engineering and Applications, P271, DOI 10.3778/j.issn.1002-8331.2108-0346
[13]  
Liu Y., 2019, INT SOC DESIGN CONF, V36, P280
[14]  
Lu X., 2017, IEEE T INTELL TRANSP, V18, P3473
[15]   Mini-YOLOv3: Real-Time Object Detector for Embedded Applications [J].
Mao, Qi-Chao ;
Sun, Hong-Mei ;
Liu, Yan-Bo ;
Jia, Rui-Sheng .
IEEE ACCESS, 2019, 7 :133529-133538
[16]  
Meng D., 2020, IEEE INT CON AUTO SC, V198, P103015
[17]  
[邱联奎 Qiu Liankui], 2014, [合肥工业大学学报. 自然科学版, Journal of Hefei University of Technology. Natural Science], V37, P572
[18]   Artificial intelligence (AI) in augmented reality (AR)-assisted manufacturing applications: a review [J].
Sahu, Chandan K. ;
Young, Crystal ;
Rai, Rahul .
INTERNATIONAL JOURNAL OF PRODUCTION RESEARCH, 2021, 59 (16) :4903-4959
[19]  
Simonyan K, 2015, Arxiv, DOI arXiv:1409.1556
[20]   Secrets of Optical Flow Estimation and Their Principles [J].
Sun, Deqing ;
Roth, Stefan ;
Black, Michael J. .
2010 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2010, :2432-2439