Training

• Download the Datasets

• Train the model with default arguments by running

python train.py

Evaluation

1. Download the model and place it in ./pretrained_models/

2. Download test datasets (Test100, Rain100H, Rain100L, Test1200, Test2800) from here and place them in ./Datasets/Synthetic_Rain_Datasets/test/

3. Run

python test.py

To reproduce PSNR/SSIM scores of the paper, run

evaluate_PSNR_SSIM.m