Channel Consistency Prior and Self-Reconstruction Strategy Based Unsupervised Image Deraining
It addresses the problem of poor generalization in real-world image deraining for computer vision applications, offering an incremental improvement over existing unsupervised methods.
The paper tackles unsupervised image deraining by proposing a framework that generates pseudo clean and rainy image pairs from unpaired data, using a channel consistency loss and self-reconstruction strategy to improve performance and generalization. Experiments show it surpasses state-of-the-art unsupervised methods on synthetic and real-world datasets.
Recently, deep image deraining models based on paired datasets have made a series of remarkable progress. However, they cannot be well applied in real-world applications due to the difficulty of obtaining real paired datasets and the poor generalization performance. In this paper, we propose a novel Channel Consistency Prior and Self-Reconstruction Strategy Based Unsupervised Image Deraining framework, CSUD, to tackle the aforementioned challenges. During training with unpaired data, CSUD is capable of generating high-quality pseudo clean and rainy image pairs which are used to enhance the performance of deraining network. Specifically, to preserve more image background details while transferring rain streaks from rainy images to the unpaired clean images, we propose a novel Channel Consistency Loss (CCLoss) by introducing the Channel Consistency Prior (CCP) of rain streaks into training process, thereby ensuring that the generated pseudo rainy images closely resemble the real ones. Furthermore, we propose a novel Self-Reconstruction (SR) strategy to alleviate the redundant information transfer problem of the generator, further improving the deraining performance and the generalization capability of our method. Extensive experiments on multiple synthetic and real-world datasets demonstrate that the deraining performance of CSUD surpasses other state-of-the-art unsupervised methods and CSUD exhibits superior generalization capability.