Skip to content

Latest commit

 

History

History
39 lines (32 loc) · 1.92 KB

README.md

File metadata and controls

39 lines (32 loc) · 1.92 KB
Raw

Swift-SRGAN - Rethinking Super-Resolution for real-time inference

This repository is the official implementation of the paper "Swift-SRGAN - Rethinking Super-Resolution for real-time inference" https://arxiv.org/abs/2111.14320

Architecture

Super-Resolution Examples

All images on the left side are the original high resolution images and images on the right side are the 4x super-resolution output from our model.

Pre-trained Models

Check the releases tab for pre-trained 4x and 2x upsampling generator models

Training

  1. install requirements with: pip install -r requirements.txt

  2. Train the model by executing:

    cd swift-srgan
python train.py --upscale_factor 4 --crop_size 96 --num_epochs 100

  3. To convert the generator model to torchscript, run

python optimize-graph.py --ckpt_path ./checkpoints/netG_4x_epoch100.pth.tar --save_path ./checkpoints/optimized_model.pt --device cuda

Please cite our article

 @article{krishnan2021swiftsrgan,
 title={SwiftSRGAN--Rethinking Super-Resolution for Efficient and Real-time Inference},
 author={Krishnan, Koushik Sivarama and Krishnan, Karthik Sivarama},
 journal={arXiv preprint arXiv:2111.14320},
 year={2021}
 }