Implement VDSR with TensorFlow

GitHub license


Introduction

We present a highly accurate single-image super-resolution (SR) method, learn residuals only and use extremely high learning rates enabled by adjustable gradient clipping


Environment


Dependency

pip

If you meet the problem with opencv when run the program

libSM.so.6: cannot open shared object file: No such file or directory

please install dependency package

sudo apt-get install libsm6
sudo apt-get install libxrender1

How to train

python main.py

if you want to see the flag

python main.py -h

trainning with 10 layer (default is 20)

python main.py --layer 10

How to test

If you don't input a Test image, it will be default image

python main.py --is_train False

then result will put in the result directory

if you want to see the result with custom layer ,ex: 10 layer

python main.py --is_train False --layer 10

If you want to Test your own iamge

use test_img flag

python main.py --is_train False --test_img Train/t20.bmp

then result image also put in the result directory


Result


Reference

kweisamx/SRCNN