Counting 1,868 Big Data & Machine Learning Frameworks, Toolsets, and Examples...
Suggestion? Feedback? Tweet @stkim1

Author
Last Commit
Oct. 14, 2017
Created
Sep. 13, 2017

image

gans-collection.torch

Torch implementation of various types of GANs (e.g. DCGAN, ALI, Context-encoder, DiscoGAN, CycleGAN, EBGAN). Note that EBGAN and BEGAN implementation is still not stable yet. I am working on this.

image

Contents

Prerequisites

  • Torch7
  • python2.7
  • cuda
  • other torch packages (display, hdf5, image ...)

Usage

  1. download training data:
python download.py --datasets <dataset>
(e.g) python run.py --datasets celebA

---------------------------------------
The training data folder should look like : 
<train_data_root>
                |--classA
                        |--image1A
                        |--image2B ...
                |--classB
                        |--image1B
                        |--image2B ...
---------------------------------------
  1. run GANs training:
    Note that you need to change parameter options in "script/opts.lua" for each GANs.
python run.py --type <gan_type>
(e.g) python run.py --type dcgan

Display GUI : How to see generated images in real-time?

step by step instruction:

1. set server-related options(ip, port, etc.) in "script.opts.lua"
2. run server (python server.py --type <gan_type>)
3. open web browser, and connect. (https://<server_ip>:<server_port>)

you will see like this: image

Results

training Final

Acknowledgement

Author

MinchulShin, @nashory
Will keep updating other types of GANs.
Any insane bug reports or questions are welcome. (min.stellastra[at]gmail.com) :-)