A couple of simple GANs in Keras
Switch branches/tags
Nothing to show
Clone or download
Latest commit 0170147 Jan 28, 2018

README.md

WARNING!!!!

This repository is not maintained! I highly recommend you use a better maintained and up to date library such as: https://github.com/eriklindernoren/Keras-GAN


KerasGAN

This module includes a GAN implementation in Keras for the MNIST data set See full article @ https://oshearesearch.com/index.php/2016/07/01/mnist-generative-adversarial-model-in-keras/

GAN Overview

The GAN includes a generative and discrimintive network defined in Keras' functional API, they can then be chained together to make a composite model for training end-to-end. GAN BlockDiag

Generated Images

Generated Images aren't perfect, the network is still pretty small and additional tuning would likely help. Generated Digits

Learning Rates

I tend to find the having a larger (faster) learning rate on the discrimintive model leads to better results than keeping them equal in the discriminitive and generative training tasks. Would be curious to hear from others who are familiar with GAN tuning here. When training with imbalanced learning rates like this, discriminitive loss stays pretty low, and the discriminitive model generally stays ahead of discriminatring new strange represenentations from the generative model. Training Loss