Skip to content

A tensorflow implementation of informative generative adversarial network (InfoGAN ). I've added supervised loss to original InfoGAN.

License

Notifications You must be signed in to change notification settings

buriburisuri/supervised_infogan

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Supervised InfoGAN

A tensorflow implementation of Xi Chen et al's "InfoGAN: Interpretable Representation Learning by Information Maximizing Generative Adversarial Nets" paper. ( See : https://arxiv.org/abs/1606.03657 ) I've added supervised loss to original InfoGAN to achieve the consistency of generated categories and training stability. The result is promising. It helped the model to generate consistent categories and to converge fast compared to original InfoGAN.

Dependencies

  1. tensorflow >= rc0.10
  2. sugartensor >= 0.0.1

Training the network

Execute


python train.py

to train the network. You can see the result ckpt files and log files in the 'asset/train' directory. Launch tensorboard --logdir asset/train/log to monitor training process.

Generating image

Execute


python generate.py

to generate sample image. The 'sample.png' file will be generated in the 'asset/train' directory.

Generated image sample

This image was generated by Supervised InfoGAN network.

And this image was generated by category factors.

And this image was generated by continuous factors.
You can see the rotation change along the X axis and the thickness change along the Y axis.

The following image is the loss chart in the training process. This looks more stable than my original InfoGAN implementation.

Other resources

  1. Original GAN tensorflow implementation
  2. InfoGAN tensorflow implementation
  3. EBGAN tensorflow implementation

Authors

Namju Kim (buriburisuri@gmail.com) at Jamonglabs Co., Ltd.

About

A tensorflow implementation of informative generative adversarial network (InfoGAN ). I've added supervised loss to original InfoGAN.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages