search

swz30 / MPRNet

[CVPR 2021] Multi-Stage Progressive Image Restoration. SOTA results for Image deblurring, deraining, and denoising.
Other
1.16k stars 188 forks source link

About the real dataset #54

Closed szpxmu closed 3 years ago

szpxmu commented 3 years ago

image I would like to compare the performance of these two real rain images, can you send me the result image?

swz30 commented 3 years ago

Hi @hellogry

You can use the following file to perform inference on your own images: https://github.com/swz30/MPRNet/blob/main/demo.py

Here is an example for Deraining: python demo.py --task Deraining --input_dir ./samples/input/ --result_dir ./samples/output/