The invention discloses a two-stage progressive
rain removal method and
system, and aims to solve the problem that the
vehicle detection precision of an intelligent
connected vehicle is reduced in a rainy day scene. According to the method, a special
vehicle detection data set MCGVD is constructed, the MCGVD is divided into a
training set and a
test set, and a two-stage progressive
rain removal network TSPNet including an improved U-Net
feature extraction network and a
feature fusion mechanism is used for training. Inputting the
test set into the trained TSPNet, performing channel dimension raising through 1 * 1
convolution, extracting shallow rain stripe information, extracting deep rain stripe information through an improved
encoder-
decoder architecture, and finally recovering to obtain a vehicle image, thereby realizing rainy day
vehicle detection. According to the
system, through the lightweight
feature extraction and weighting module and the high-efficiency feature transmission and fusion module, the
rain removal effect and the
image quality are improved, the performance of the vehicle detection
system under complex weather conditions is remarkably improved, and the system has important practical application value.