ElefHead / modular-gan

Keras model subclassing implementation of a GAN. Based on https://twitter.com/fchollet/status/1250622989541838848

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Modular GAN

A DCGAN implemented with keras model subclassing.
Implemented because I wanted to replicate François Chollet's tweet. Uses MNIST dataset.

Usage

To run training :

python -m gan.main --mode=train --save '{"batch_size": 32, "epochs": 50, "dataset": "mnist", "latent_dim": 100, "buffer_size": 60000}'

Once model is trained and saved, modify the following for running evaluation :

python -m gan.main --mode=evaluate --save '{"generator_path": "<model_save_path>", "latent_dim": 100, "num_images": 20}'

About

Keras model subclassing implementation of a GAN. Based on https://twitter.com/fchollet/status/1250622989541838848


Languages

Language:Jupyter Notebook 84.9%Language:Python 14.8%Language:Shell 0.4%