Skip to content

A tensorflow implementation of VDSR with PReLU

Notifications You must be signed in to change notification settings

lygztq/pVDSR

 
 

Repository files navigation

Image Super-Resolution Using Deep Concolutional Neural Networks with PReLU


Experimental Setup

Firstly, this project was implemented using Python 2.7, so at the begining you should check your Python version.

The Python package needed:

Secondly, we use MATLAB(Linux version) to preprocess the training data(because when we generate the training data from image, using MATLAB is much faster than using openCV on Python). So make sure you have already installed MATLAB on your server.

Finally, you can get the training data from: data


How To Use

preprocessing data

  • Download the training and test data from here
  • Unzip the data.zip, unzip 291.zip and Set14.zip inside.
  • Copy the 291 and Set14 directory to "/data" directory.
  • Run generate_train.m and generate_test.m to generate training and test data.

Training

Run "train.py" to train. If you want to start from a checkpoint, using "py -2 train.py --model_path [your checkpoint path]"

Testing

Run "python test.py"

Plot Result

Run "python plot.py"

Use to increase resolution of images

Run python increase_resolution.py --img [your-image-path]


A Test Model

If you want to use the network without training, download this model, the result may be not very good since it just a test model.

About

A tensorflow implementation of VDSR with PReLU

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 67.9%
  • MATLAB 32.1%