Awesome

IDN-tensorflow

[Original Caffe version]

Testing

• Install Tensorflow 1.11, Matlab R2017a
• Download Test datasets
• Modify config.py (if you want to test x3 model on Set14, config.TEST.model_path = 'checkpoint_x3/model.ckpt' config.TEST.dataset = 'Set14') and test.py (scale = 3).
• Run testing:

python test.py

Training

• Download Training dataset
• Modify config.py (if you want to train x4 model, config.TRAIN.hr_img_path = '/path/to/DIV2K_train_HR/' config.TRAIN.checkpoint_dir = 'checkpoint_x4/' config.VALID.hr_img_path = '/path/to/DIV2K_valid_HR/' config.VALID.lr_img_path = '/path/to/DIV2K_valid_LR_x4/') and train_SR.py (scale = 4)
• Run training:

python train_SR.py

Note

This TensorFlow version is trained with DIV2K training dataset on RGB channels. Additionally, We modify the upsample layer to subpixel convolution (the original version is transposed convolution).

Results

Test_results

The following PSNR/SSIMs are evaluated on Matlab R2017a and the code can be referred to Evaluate_PSNR_SSIM.m.

Model Parameters

Citation

If you find IDN useful in your research, please consider citing:

@inproceedings{Hui-IDN-2018,
  title={Fast and Accurate Single Image Super-Resolution via Information Distillation Network},
  author={Hui, Zheng and Wang, Xiumei and Gao, Xinbo},
  booktitle={CVPR},
  pages = {723--731},
  year={2018}
}