Skip to content

hhcs9527/BRMv2-MPRNet

Repository files navigation

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/

Testing on GoPro dataset

  • Download images of GoPro and place them in ./Datasets/GoPro/test/
  • Run
python test.py --dataset GoPro

Testing on HIDE dataset

  • Download images of HIDE and place them in ./Datasets/HIDE/test/
  • Run
python test.py --dataset HIDE

Testing on RealBlur-J dataset

  • Download images of RealBlur-J and place them in ./Datasets/RealBlur_J/test/
  • Run
python test.py --dataset RealBlur_J

Testing on RealBlur-R dataset

  • Download images of RealBlur-R and place them in ./Datasets/RealBlur_R/test/
  • Run
python test.py --dataset RealBlur_R

To reproduce PSNR/SSIM scores of the paper on GoPro and HIDE datasets, run this MATLAB script

evaluate_GOPRO_HIDE.m 

To reproduce PSNR/SSIM scores of the paper on RealBlur dataset, run

evaluate_RealBlur.m 

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published