srgan-tensorflow2.0
SRGAN implementation using Tensorflow 2.0
Download other folders (samples, models, DIV2K, archive) at:
https://drive.google.com/drive/folders/1-C3y8Ae_4NbFjrymxfF2BExlhtCzfs-2?usp=sharing
Usage:
- Change model's parameters in
config.py - How to use
train.py
- Warm-up Generator:
python train.py --mode=warmup
- Train from beginning:
python train.py
- Continue training:
python train.py --mode=continue
- Generate demo super-resolution image:
python train.py --mode=evaluate - Losses and PSNR, SSIM scores can be calculated in
evaluation.ipynb - Web deployment in
deployment