Skip to content

lookway/RDN-TensorFlow

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

25 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

RDN-TensorFlow

A TensorFlow implementation of CVPR 2018 paper Residual Dense Network for Image Super-Resolution.
Official implementation: Torch code for our CVPR 2018 paper "Residual Dense Network for Image Super-Resolution" (Spotlight)

Prerequisites

  • Python-2.7(Python-3.5 for branch python3)
  • TensorFlow-1.10.0
  • Numpy-1.14.5
  • OpenCV-2.4.9.1
  • h5py-2.6.0

Usage

Prepare data

Download DIV2K training data from here.
Extract and place all the images in RDN-TensorFlow/Train/DIV2K_train_HR.

Train

python main.py

Test

python main.py --is_train=False

Notice

If you want to use the resize function in MATLAB when generating training data and testing images as the pretrained model used, you need to install MATLAB API for Python, and run the script with option --matlab_bicubic=True.

If you want to take an original image as the input of RDN directly, you could run the script like python main.py --is_train=False --is_eval=False --test_img=Test/Set5/butterfly_GT.bmp.

References

About

A TensorFlow implementation of CVPR 2018 paper "Residual Dense Network for Image Super-Resolution".

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Python 100.0%