Awesome

【CVPR'2024🔥】Bidirectional Multi-Scale Implicit Neural Representations for Image Deraining

Bidirectional Multi-Scale Implicit Neural Representations for Image Deraining

Xiang Chen, Jinshan Pan, Jiangxin Dong

Nanjing University of Science and Technology

Primary contact: Xiang Chen (chenxiang@njust.edu.cn)

📣 News

• [24-04-03] The paper is available here.
• [24-03-20] The source codes, models and results are available.
• [24-02-27] Our paper has been accepted to CVPR 2024.

📖 Research Website: [link] (A Survey on Image Deraining)

📌 Overview

🔑 Setup

Type the command:

pip install -r requirements.txt

Install warmup scheduler

cd pytorch-gradual-warmup-lr; python setup.py install; cd ..

🧩 Dataset Preparation

🛠️ Training and Testing

1. Please download the corresponding datasets and put them in the folder Datasets/.
2. Follow the instructions below to begin training our model.

bash train.sh

Run the script then you can find the generated experimental logs in the folder checkpoints.

3. Follow the instructions below to begin testing our model.

python test.py

Run the script then you can find the output visual results in the folder results/.

🤖 Pre-trained Models

🚨 Performance Evaluation

See folder "evaluations"

1. for Rain200L/H and SPA-Data datasets: PSNR and SSIM results are computed by using this Matlab Code.

2. for DID-Data and DDN-Data datasets: PSNR and SSIM results are computed by using this Matlab Code.

🚀 Visual Deraining Results

🔎 More Results

👍 Acknowledgement

Thanks for their awesome works (DeepRFT and NeRCo).

📘 Citation

Please consider citing our work as follows if it is helpful.

@InProceedings{NeRD-Rain,
    author={Chen, Xiang and Pan, Jinshan and Dong, Jiangxin}, 
    title={Bidirectional Multi-Scale Implicit Neural Representations for Image Deraining},
    booktitle={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
    month={June},
    year={2024}
}