Laser & Optoelectronics Progress, Volume. 60, Issue 22, 2210006(2023)

Two-Stage Progressive Image Deraining Algorithm for Vehicle Detection in Rainy Days

Daifang Hu1, Qiuhong Tong1、*, Guoqing Chai1, Kai Wang2, Yuwei Mu1, and Shengjun Su1
Author Affiliations
  • 1School of Automobile, Chang'an University, Xi'an 710018, Shaanxi , China
  • 2School of Information Engineering, Chang'an University, Xi'an 710018, Shaanxi , China
  • show less

    This study proposes a two-stage progressive image deraining algorithm for vehicle detection in rainy days. The proposed algorithm aims to improve the accuracy of vehicle detection in rainy days and solve the problem of accuracy degradation caused by rain streak interference in the vehicle detection system of intelligent and connected vehicles. For the algorithm, a two-stage progressive deraining network was developed with light feature extraction and weighting block along with efficient feature transfer and fuse block as the core to realize the mining and capture of rain streak information and achieve accurate deraining. The deraining images were input to benchmark vehicle detector YOLOv5 for verifying the effectiveness of the proposed algorithm. Furthermore, a mix vehicle dataset was constructed based on the working environment of intelligent and connected vehicles. The gains of the proposed deraining algorithm on the precision, recall, and mAP@0.5 of the benchmark vehicle detector YOLOv5 are 3.0 percentage points, 8.9 percentage points, and 7.6 percentage points, respectively, under a rainy traffic scenario compared with other algorithms. The results prove that the proposed deraining algorithm considerably improves the accuracy of vehicle detection in rainy days and can be used in practice.

    Tools

    Get Citation

    Copy Citation Text

    Daifang Hu, Qiuhong Tong, Guoqing Chai, Kai Wang, Yuwei Mu, Shengjun Su. Two-Stage Progressive Image Deraining Algorithm for Vehicle Detection in Rainy Days[J]. Laser & Optoelectronics Progress, 2023, 60(22): 2210006

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Image Processing

    Received: Apr. 7, 2023

    Accepted: Jun. 20, 2023

    Published Online: Nov. 6, 2023

    The Author Email: Tong Qiuhong (tongqiuhong@chd.edu.cn)

    DOI:10.3788/LOP231053

    Topics