Skip to content

luke-dinh/Image-Super-Resolution

Repository files navigation

Image-Super-Resolution

Image Super-Resollution using Deep Learning, implemented in PyTorch, with an API for easily usage.

1. Orriginal model

  • SRGAN Architecture

2. Proposed method:

The whole framework: A GAN-based model (mSRGAN) is used to train the model. After training, the Generator block (G block) is used for API to create new higher resolution image from the lower one.

We modified the G block architecture from the previous model for better training and lowwer weights. The architecture of the block is shown as below (for training and usage):

3. Usage

  • For non-API usage:

Using with images:

pip3 install requirements.txt
cd mSRGAN
python3 test_image.py --image_name = "YOUR IMAGE PATH"

Using with video:

pip3 install requirements.txt
cd mSRGAN
python3 test_video.py --video_name = "YOUR VIDEO PATH"
  • Using via API:

python app.py 

IP Browser: 0.0.0.1:5000