Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.
You can not select more than 25 topics
Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
tiantian132
a3342d8119
|
1 year ago | |
|---|---|---|
| .. | ||
| Datasets | 1 year ago | |
| pretrained_models | 1 year ago | |
| utils | 1 year ago | |
| MPRNet.py | 1 year ago | |
| README.md | 1 year ago | |
| config.py | 1 year ago | |
| data_RGB.py | 1 year ago | |
| dataset_RGB.py | 1 year ago | |
| evaluate_PSNR_SSIM.m | 1 year ago | |
| losses.py | 1 year ago | |
| test.py | 1 year ago | |
| train.py | 1 year ago | |
| training.yml | 1 year ago | |
README.md
Training
-
Download the Datasets
-
Train the model with default arguments by running
python train.py
Evaluation
-
Download the model and place it in
./pretrained_models/ -
Download test datasets (Test100, Rain100H, Rain100L, Test1200, Test2800) from here and place them in
./Datasets/Synthetic_Rain_Datasets/test/ -
Run
python test.py
To reproduce PSNR/SSIM scores of the paper, run
evaluate_PSNR_SSIM.m