Improved YOLOx-based vehicle detection method for low light environment

被引:0
作者
Yang, Xiaohan [1 ]
Wang, Jun [2 ]
Duan, Zhongxing [1 ]
Hui, Leile [3 ]
机构
[1] Xian Univ Architecture & Technol, Coll Informat & Control Engn, Xian 710055, Peoples R China
[2] Traff Engn Construct Bur Jiangsu Prov, Nanjing 210024, Peoples R China
[3] China Northwest Architecture Design & Res Inst Co, Xian 710018, Peoples R China
基金
中国国家自然科学基金;
关键词
low-light environment; swin-transformer; vehicle detection; image enhancement; attention mechanism;
D O I
10.37188/CJLCD.2023-0166
中图分类号
O7 [晶体学];
学科分类号
0702 ; 070205 ; 0703 ; 080501 ;
摘要
For the low light environment such as road tunnels, the acquired vehicle images are affected by external factors, which leads to low vehicle detection accuracy. For this problem, a real-time vehicle detection method for low light environment with improved YOLOx algorithm is proposed. Firstly, the collected vehicle images are enhanced based on the guiding filter and regional energy characteristic fusion criterion to solve the problems of uneven illumination and blurred target contour information in the images.Secondly, based on the Swin-Transformer network structure, the backbone network of the vehicle detection algorithm with improved YOLOx is constructed, and the global modeling capability of Transformer is used to encode the key semantic information in the images and strengthen the extraction capability of the network detail features. Meanwhile, recursive gated convolution is introduced to replace the null convolution in the neck network to improve the network's high-level semantic modeling capability. Finally, a convolutional attention mechanism is introduced to enhance the network's extraction and fusion of key features for low-illumination images. Experimental validation on the constructed tunnel vehicle detection dataset UA-DETTUN shows that the proposed method achieves an average detection accuracy of 96.1%,which is 6.5% better than the YOLOx algorithm before improvement, while the detection speed of the network meets the requirement of real-time detection. The proposed method has high application value in vehicle detection.
引用
收藏
页码:801 / 812
页数:12
相关论文
共 27 条
[11]   基于改进Yolov4的车辆目标检测 [J].
李松江 ;
耿兰兰 ;
王鹏 .
计算机工程, 2023, 49 (04) :272-280
[12]  
[李志远 Li Zhiyuan], 2021, [小型微型计算机系统, Journal of Chinese Computer Systems], V42, P2572
[13]   SSD: Single Shot MultiBox Detector [J].
Liu, Wei ;
Anguelov, Dragomir ;
Erhan, Dumitru ;
Szegedy, Christian ;
Reed, Scott ;
Fu, Cheng-Yang ;
Berg, Alexander C. .
COMPUTER VISION - ECCV 2016, PT I, 2016, 9905 :21-37
[14]   Swin Transformer: Hierarchical Vision Transformer using Shifted Windows [J].
Liu, Ze ;
Lin, Yutong ;
Cao, Yue ;
Hu, Han ;
Wei, Yixuan ;
Zhang, Zheng ;
Lin, Stephen ;
Guo, Baining .
2021 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2021), 2021, :9992-10002
[15]  
Rao Yuchen, 2022, Advances in Neural Information Processing Systems
[16]   YOLO9000: Better, Faster, Stronger [J].
Redmon, Joseph ;
Farhadi, Ali .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :6517-6525
[17]   You Only Look Once: Unified, Real-Time Object Detection [J].
Redmon, Joseph ;
Divvala, Santosh ;
Girshick, Ross ;
Farhadi, Ali .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :779-788
[18]  
Redmon Joseph, 2018, CORR ABS180402767
[19]  
Tolstikhin I, 2021, ADV NEUR IN, V34
[20]  
[涂媛雅 Tu Yuanya], 2023, [小型微型计算机系统, Journal of Chinese Computer Systems], V44, P211