puffnjackie / pytorch-super-resolution-implementations

Image super-resolution implementations

PyTorch super-resolution implementations

A collections of Super-resolution models

Requirements

• python 3.6
• numpy
• pytorch >= 0.4.0

Models

• [RDN] https://arxiv.org/abs/1802.08797

Usage

• training

python3 train.py --upscale_factor 2 --datapath /model/path/to/folder

• super-resolve

python3 inference.py --input_image /path/to/img --output_filename /path/to/img --model /path/to/pretrained/model

Reference

• https://github.com/pytorch/examples
• https://github.com/andreasveit/densenet-pytorch
• https://github.com/xternalz/WideResNet-pytorch