/EDSR-tensorflow2

EDSR implementation in tensorflow 2

Primary LanguagePythonMIT LicenseMIT

EDSR-tensorflow2

Tensorflow 2 implementation of Enhanced Deep Residual Networks for Single Image Super-Resolution(CVPR2017)

EDSR

Usage

$ python main.py [-h] N_TRAIN_DATA N_TEST_DATA N_RES_BLOCK BATCH_SIZE EPOCHS SCALE

DIV2K example

$ python main.py 800 100 32 16 200 4

Prerequisites

  • Python 3.7
  • Tensorflow 2
  • Numpy

Directory

EDSR-tensorflow2
├── main.py              // main program
├── model.py             // edsr model
├── data_generator.py    // data augmentation
└── utils.py             // psnr, mae

Sample Results

  • trained by Urban100
  • test image set is Set14

sample