Paper Info
Title
MSFSA-GAN: Multi-Scale Fusion Self Attention Generative Adversarial Network for Single Image Deraining
Abstract
Bad weather such as rainy days will seriously affect the image quality and the accuracy of visual processing algorithm. In order to improve the image deraining quality, a multi-scale fusion self attention generation adversarial network (MSFSA-GAN) is proposed. This network uses different scales to extract input characteristics of rain lines. First, Gaussian pyramid rain maps with different scales are generated by Gaussian algorithm. Then, in order to extract the features of rain lines with different scales, the coarse fusion module and fine fusion module are designed respectively. Next, the extracted features are fused at different scales. In this process, the self attention mechanism is introduced to make the network focus on the extracted features of different scales. And before the fusion, the rain pattern reconstruction operation is also carried out, so that the network can reproduce the input image more perfectly. Finally, it is input into the discriminator network with dense blocks to obtain the image that removes the rain lines. We used R100H and R100L datasets to train and test our network. The results show that our method as high as 27.79 in PSNR and UQI is 0.94, which is superior to the existing methods in performance. Meanwhile, we also compared the cost of time, the result of our network is only 0.02s.
Year
DOI
Venue
2022
10.1109/ACCESS.2022.3162224
IEEE ACCESS
Keywords
DocType
Volume
Rain, Feature extraction, Generators, Generative adversarial networks, Deep learning, Data mining, Task analysis, Rain removal, MSFSA-GAN, self attention, dense block
Journal
10
ISSN
Citations 
PageRank 
2169-3536
0
0.34
References 
Authors
0
6
Name
Order
Citations
PageRank
Wang Xue100.34
Cheng Huan-Xin200.34
Sun Sheng-Yi300.34
Jiang Ze-Qin400.34
Kai Cheng53912.36
Cheng Li627939.13