Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Bidirectional Multi-Scale Implicit Neural Representations for Image Deraining

About

How to effectively explore multi-scale representations of rain streaks is important for image deraining. In contrast to existing Transformer-based methods that depend mostly on single-scale rain appearance, we develop an end-to-end multi-scale Transformer that leverages the potentially useful features in various scales to facilitate high-quality image reconstruction. To better explore the common degradation representations from spatially-varying rain streaks, we incorporate intra-scale implicit neural representations based on pixel coordinates with the degraded inputs in a closed-loop design, enabling the learned features to facilitate rain removal and improve the robustness of the model in complex scenarios. To ensure richer collaborative representation from different scales, we embed a simple yet effective inter-scale bidirectional feedback operation into our multi-scale Transformer by performing coarse-to-fine and fine-to-coarse information communication. Extensive experiments demonstrate that our approach, named as NeRD-Rain, performs favorably against the state-of-the-art ones on both synthetic and real-world benchmark datasets. The source code and trained models are available at https://github.com/cschenxiang/NeRD-Rain.

Xiang Chen, Jinshan Pan, Jiangxin Dong• 2024

Related benchmarks

TaskDatasetResultRank
Image DerainingRain100L
PSNR42
249
Image DerainingRain100H
PSNR32.86
52
Image DerainingSPA-Data
PSNR48.9
45
Image DerainingRealRain1K L
PSNR38.64
40
Image DerainingSPA-Data (test)
PSNR33.03
24
Image DerainingRain200L
PSNR41.3
23
Image DerainingDID-Data
PSNR35.36
15
Image DerainingNight-Rain
PSNR38.42
15
Image DerainingRain800
PSNR28.82
13
Image DerainingRealRain1K-H
PSNR36.69
13
Showing 10 of 25 rows

Other info

Follow for update