nashory / gans-collection.torch

Torch implementation of various types of GAN (e.g. DCGAN, ALI, Context-encoder, DiscoGAN, CycleGAN, EBGAN, LSGAN)

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.

Contents

• DCGAN
• Context-encoder
• ALI
• DiscoGAN
• CycleGAN
• Energy-Based GAN (EBGAN)
• LSGAN

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 ...
---------------------------------------

2. 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:

Results

training	Final

Acknowledgement

• brought dataloader code from (DCGAN)
• referenced the code from (Context-encoder)

Author

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