Cycle-attention-derain: unsupervised rain removal with CycleGAN

Miaomiao Chen,Pei Wang,Dehai Shang,Pengjie Wang
DOI: https://doi.org/10.1007/s00371-023-02947-2
2023-06-25
Abstract:Single image deraining is a fundamental task in computer vision, which can greatly improve the performance of subsequent high-level tasks under rainy conditions. Existing data-driven rain removal methods rely heavily on paired training data, which is expensive to collect. In this paper, we propose a new unsupervised method, called Cycle-Attention-Derain, which removes rain from single images in the absence of paired data. Our method is based on the CycleGAN framework with two major novelties. First, since rain removal is highly correlated with analyzing the texture features of an input image, we propose a novel attention fusion module (AFM) with complementary channel attention and spatial attention, which can effectively learn more discriminative features for rain removal. Second, to further improve the generalization ability of our model, we propose a global–local attention discriminator architecture with an attention mechanism to guide the network training, so that the rain removal results are realistic, both globally and locally. Our proposed model is able to remove rain streaks and raindrops of varying degrees without paired training images. Extensive experiments on synthetic and real datasets demonstrate that the proposed method outperforms most of the state-of-the-art unsupervised rain removal methods in terms of both PSNR and SSIM on Rain800 datasets and achieves slightly close results to other popular supervised learning methods.
computer science, software engineering
What problem does this paper attempt to address?