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 条
[1]   Feature-Fused SSD: Fast Detection for Small Objects [J].
Cao, Guimei ;
Xie, Xuemei ;
Yang, Wenzhe ;
Liao, Quan ;
Shi, Guangming ;
Wu, Jinjian .
NINTH INTERNATIONAL CONFERENCE ON GRAPHIC AND IMAGE PROCESSING (ICGIP 2017), 2018, 10615
[2]   Yolo V4 for Advanced Traffic Sign Recognition With Synthetic Training Data Generated by Various GAN [J].
Dewi, Christine ;
Chen, Rung-Ching ;
Liu, Yan-Ting ;
Jiang, Xiaoyi ;
Hartomo, Kristoko Dwi .
IEEE ACCESS, 2021, 9 :97228-97242
[3]  
Feng Z., 2021, PROF DEV EDUC, V15, P280
[4]   CA-Net: Comprehensive Attention Convolutional Neural Networks for Explainable Medical Image Segmentation [J].
Gu, Ran ;
Wang, Guotai ;
Song, Tao ;
Huang, Rui ;
Aertsen, Michael ;
Deprest, Jan ;
Ourselin, Sebastien ;
Vercauteren, Tom ;
Zhang, Shaoting .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2021, 40 (02) :699-711
[5]   GhostNet: More Features from Cheap Operations [J].
Han, Kai ;
Wang, Yunhe ;
Tian, Qi ;
Guo, Jianyuan ;
Xu, Chunjing ;
Xu, Chang .
2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2020, :1577-1586
[6]   Spatial Pyramid Pooling in Deep Convolutional Networks for Visual Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2015, 37 (09) :1904-1916
[7]  
Jiang HZ, 2017, IEEE INT CONF AUTOMA, P650, DOI [10.1109/MWSYM.2017.8058653, 10.1109/FG.2017.82]
[8]   Moving object detection using median-based scale invariant local ternary pattern for video surveillance system [J].
Kalirajan, K. ;
Sudha, M. .
JOURNAL OF INTELLIGENT & FUZZY SYSTEMS, 2017, 33 (03) :1933-1943
[9]  
Kuznetsova A, 2020, LECT NOTES COMPUT SC, V12557, P233, DOI 10.1007/978-3-030-64221-1_20
[10]  
Lai L., 2016, INFRARED LASER ENG, V45, P280