Giter VIP home page Giter VIP logo

super_resolution's Introduction

This repo contains implementations of papers

Super Resolution

Two methods are being used to perform super resolution

  • EDSR + Pixel Shuffling + Perceptual Loss
  • Pre-trained GAN (traning Generator and critic seperately and then training both)

Requirements

  • Pytorch>=0.3
  • Fastai

Usage

$ python super.py

Dataset

  • ImageNet 10% sample
  • Celeb A

EDSR

Requires Fastai==0.7 (super_res_edsr.py)

In this method, We're making use of Pixel shuffle along with EDSR (Removal of BN in skip connection)

Original image

Prediction 

Input image

Prediction 

.

Pre Trained GANs

PreRequisites

  • FastAI>=1.45
  • Pytorch >= 1.0

Creating Low quality dataset

Use crappification from (utils.py)

We use a Unet Generator which has Resnet34/50 (Pretrained) as it's head (modifiable) Generator is trained with low quality images with ground truth being images from original dataset.

When pretraining generator only, here's the output:

Input----------------------------------------------------Prediction------------------------------------------Target

gen_pre1 gen_pre2 gen_pre3

Critic is trained on generator images and original images

Usage

Check the path variables carefully

$ python3 super_res_gan.py 

References

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.