Skip to content
/ SISR Public

Capstone project for a course on machine learning and deep learning - Single Image Super Resolution

Notifications You must be signed in to change notification settings

jerome-p/SISR

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

15 Commits
 
 
 
 
 
 

Repository files navigation

Deep learning method for Single Image Super Resolution

Capstone project for a course on machine learning and deep learning

  • This is an implementation of Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial Network.Used left-thomas's implementation of SRGAN.

  • Trained a modified SRGAN model using google colab, with the VOC2012 dataset(approx 1000 images) for different upscale values.

  • Tried to modify the network to be able to take in any upscale value (not fractional), instead having only exponentials of 2 as is in the paper, upon doign so, noticed that the modification resulted in more number of parameters than the original model, for upscale values like 4,6,8 etc.

  • This is why I think it resulted in more number of parameters: explanation

tl;dr

Original model uses lesser number of "channels" to upscale the image when compared to the modified model. Read this to understand more


Some results from the trained model

Modified SRGAN model, trained for 930 epochs:

input : input_flower

output: output_flower PSNR:27.07


input : input_dog

output: output_dog PSNR:20.50


input : input_house

output: output_house PSNR:28.57


Pre-trained ESRGAN:

input : input_ESRGAN

output: output_ESRGAN

About

Capstone project for a course on machine learning and deep learning - Single Image Super Resolution

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published