/ImageSR-Pytorch

An image super-resolution implemented by Pytorch

Primary LanguagePythonMIT LicenseMIT

ImageSR-Pytorch

MIT License

Preparation

  • Python 3.6
  • Pytorch 1.1.0

Model

  • Use a simple version EDSR for image super-resolution

Training

  • Prepare training dataset. Please download DIV2K dataset: link
  • Move the training dataset to ./DIV2K_train_HR
  • Check the config in config.yaml
  • Run the script python train.py config.yaml

Inference

  • Check the config in config.yaml
  • Run the script python inference.py config.yaml

Reference