- Notifications
You must be signed in to change notification settings - Fork2
purvilmehta06/Image-Super-Resolution-Using-GAN-SRGAN
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
This repo contains the project work carried out for the course Deep Learning in my B.Tech Final Year DA-IICT. It is the replication of the code in simpler terms available on GitHub.
- Reference Paper Link:Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial Network
- Reference Code Link:Github
You need to have the high resolution images for training. In this project, we have used images fromDIV2K - bicubic downscaling x4 competition, so the hyper-paremeters are (like number of epochs) are seleted basic on that dataset, if you change a larger dataset you can reduce the number of epochs. Following are some links to download training andd test dataset:
We have trained our model upto 100 epochs. Weights can be found from thePre Trained Weights/
folder. You can fine tune the parameter and resume the training of the model. To load the model weights, follow the step shown below.
- Provide the path of your weights in the first cell variable
G_weights_load
andG_weights_load
. - Uncomment 2 lines i.e line no 16 and 17 in the training cell.
netG.load_state_dict(torch.load(G_weights_load))
netD.load_state_dict(torch.load(D_weights_load))
- Open ipynb file in either google colab.
- Put the notebook on GPU mode.
- Change the path depending up on your file structure in the first cell.
- Run all cells.
- Ruchit Vithani
- Bhargey Mehta
- Kushal Shah
About
This repo contains the project work carried out for the course Deep Learning in my B. Tech Final Year DA-IICT. It is the replication of the code in simpler terms available on GitHub.
Topics
Resources
Uh oh!
There was an error while loading.Please reload this page.