TA-DNN—two stage attention-based deep neural network for single image rain removal

S. Jothi Lakshmi,R. Bhuvaneshwari,P. Geetha
DOI: https://doi.org/10.1007/s11760-023-02538-7
2023-04-27
Abstract:Computer Vision algorithms easily get affected by images or videos captured in outdoor environments due to various bad weather conditions such as rain, fog, snow and haze. Two-stage deep neural network based on attention learning is proposed for single image rain removal. Based on the fact that the implicit connection among rain lines within the image is higher than that between the rain line and the background image, it is easier to learn the rain lines in the rainy image using attention learning. The proposed Two-stage Attention-based Deep Neural Network (TA-DNN) for single image rain removal essentially consists of modules such as Inception, Sequential Dual Attention Block (SDAB), and Multi-Scale Feature Aggregation Module (MSFAM) for feature extraction, rain line detection, and transfiguration, respectively. The experimental results illustrate that the proposed method performs better when compared with the state-of-the-art methods both qualitatively and quantitatively.
engineering, electrical & electronic,imaging science & photographic technology
What problem does this paper attempt to address?