ImageSR-Pytorch
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