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 条
  • [1] Bochkovskiy A., 2020, Yolov4: Optimal speed and accuracy of object detection
  • [2] [曹行健 Cao Xingjian], 2022, [中国图象图形学报, Journal of Image and Graphics], V27, P1743
  • [3] [柴勇 Chai Yong], 2009, [光学学报, Acta Optica Sinica], V29, P2732
  • [4] [杜永萍 Du Yongping], 2022, [模式识别与人工智能, Pattern Recognition and Artificial Intelligence], V35, P472
  • [5] [高德勇 Gao Deyong], 2022, [西安电子科技大学学报, Journal of Xidian University], V49, P144
  • [6] GUO L Q, 2019, Journal of Changchun Normal University, V38, P15
  • [7] He KM, 2010, LECT NOTES COMPUT SC, V6311, P1
  • [8] Multi-Focus Image Fusion Based on Discrete Walsh-Hadamard Transform and Guided Filtering
    Hu Liang
    Hu Xuejuan
    Huang Zhenhong
    Xu Lu
    Lian Lijin
    [J]. LASER & OPTOELECTRONICS PROGRESS, 2021, 58 (22)
  • [9] [赖见辉 Lai Jianhui], 2021, [公路交通科技, Journal of Highway and Transportation Research and Development], V38, P135
  • [10] [李炳林 Li Binglin], 2021, [公路交通科技, Journal of Highway and Transportation Research and Development], V38, P142