Code Monkey home page Code Monkey logo

srgan-4's Introduction

SRGAN

  • An implement of SRGAN(Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial Network) for tensorflow version.
  • In this repo, vgg19 is not used, instead, MSE is ued to train SRResNet. If you want to use vgg19 to calculate the content loss, you can download model that trained in ImageNet. Then you just need to load to your model during training phase.

Requirements

  • tensorflow 1.3.0
  • python 2.7.12 or python 3.*
  • numpy 1.13.1
  • scipy 0.17.0

Usages

downlaod repo

  • download this repo by the following instruction:

    $ git clone https://github.com/nnuyi/SRGAN.git
    $ cd SRGAN
    

download datasets

  • Firstly, you need to make some directories in the root path(in SRGAN directory)

    $ mkdir data
    $ cd data
    $ mkdir train
    $ mkdir val
    $ mkdir test   
    

train data

  • In this repo, I use parts of ImageNet datasets as train data, here you can download the datasets that I used.

  • After you have download the datasets, copy ImageNet(here I only use 3137 images) datsets to /data/train, then you have /data/train/ImageNet path, and training images are stored in /data/train/ImageNet

  • I crop image into 256*256 resolution, actually you can crop them according to your own.

val data

  • Set5 dataset is used as val data, you can download it here.

  • After you download Set5, please store it in /data/val/ , then you have /data/val/Set5 path, and val images are stored in /data/val/Set5

test data

  • Set14 dataset is used as test data, you can download it here.

  • After you download Set14, please store it in /data/test/ , then you have /data/test/Set14 path, and val images are stored in /data/test/Set14

training

  $ python main.py --is_training=True --is_testing=False

testing

  $ python main.py --is_training=False --is_testing=True

Experimental Results

Factor 4(two shuffle layers is used)

low resolution high resolution GT high resolution GEN
Alt test Alt test Alt test
Alt test Alt test Alt test
Alt test Alt test Alt test
Alt test Alt test Alt test
Alt test Alt test Alt test
Alt test Alt test Alt test

Factor 4(whole test images)

sampling image
Alt test
256*256 resolution left:GT right:GEN

References

Contacts

Email:[email protected]

srgan-4's People

Contributors

nnuyi avatar

Watchers

 avatar

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.