Skip to content

BenWare144/KDSRGAN-2018

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

KDSRGAN

An application of knowledge distillation in GAN SuperResolution

This code uses an implimentation of the SRGAN network, modified from the code of this repository SRGAN.

SRGAN Architecture

TensorFlow Implementation of "Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial Network"

Downloaded pretrained VGG layer here.

Download both HR and a set of LR images of this datset: DIV2K - bicubic downscaling x4 competition

This code uses the "unknown downscaling x4".

Run

Set your image folder and other options in config.py.

  • Train a network without knowledge distillation
python main.py
  • Start evaluation.
python main.py --mode=evaluate 
  • Train a network with knowledge distillation.
python train_distill.py

Reference

Author

License

  • For academic and non-commercial use only.
  • For commercial use, please contact [email protected].

About

Knowledge distilation applied to the SRGAN network from https://github.com/tensorlayer/srgan

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages