/DRN

Closed-loop Matters: Dual Regression Networks for Single Image Super-Resolution

Primary LanguagePythonMIT LicenseMIT

Closed-loop Matters: Dual Regression Networks for Single Image Super-Resolution

Pytorch implementation for "Closed-loop Matters: Dual Regression Networks for Single Image Super-Resolution".

Dual Regression Scheme

Dependencies

Python>=3.7, PyTorch>=1.1, numpy, skimage, imageio, matplotlib, tqdm

Quickstart (Model Testing)

Results of our pretrained models:

Model Scale #Params (M) PSNR on Set5 (dB)
DRN-S 4 4.8 32.68
8 5.4 27.41
DRN-L 4 9.8 32.74
8 10.0 27.43

You can evaluate our models on several widely used benchmark datasets, including Set5, Set14, B100, Urban100, Manga109. Note that using an old PyTorch version (earlier than 1.1) would yield wrong results.

python main.py --data_dir $DATA_DIR$ --save $SAVE_DIR$ --data_test $DATA_TEST$ --scale $SCALE$ --model $MODEL$ --pre_train $PRETRAINED_MODEL$ --test_only --save_results
  • DATA_DIR: path to save data
  • SAVE_DIR: path to save experiment results
  • DATA_TEST: the data to be tested, such as Set5, Set14, B100, Urban100, and Manga109
  • SCALE: super resolution scale, such as 4 and 8
  • MODEL: model type, such as DRN-S and DRN-L
  • PRETRAINED_MODEL: path of the pretrained model

Training Method

We use DF2K dataset (the combination of DIV2K and Flickr2K datasets) to train DRN-S and DRN-L.

python main.py --data_dir $DATA_DIR$ --save $SAVE_DIR$ --scale $SCALE$ --model $MODEL$
  • DATA_DIR: path to save data
  • SAVE_DIR: path to save experiment results
  • SCALE: super resolution scale, such as 4 and 8
  • MODEL: model type, such as DRN-S and DRN-L

Citation

If you use any part of this code in your research, please cite our paper:

@inproceedings{guo2020closed,
  title={Closed-loop Matters: Dual Regression Networks for Single Image Super-Resolution},
  author={Guo, Yong and Chen, Jian and Wang, Jingdong and Chen, Qi and Cao, Jiezhang and Deng, Zeshuai and Xu, Yanwu and Tan, Mingkui},
  booktitle={Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition},
  year={2020}
}