Skip to content

Deep Convolutional Generative Adversarial Network for MNIST Dataset

License

Notifications You must be signed in to change notification settings

stante/dcgan-mnist

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

58 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

dcgan-mnist

This repository contains an implementation of a Deep Convolutional Generative Adversarial Network (DCGAN) for MNIST. It implements the suggested architectural constraints for stable learning from the paper Unsupervised Representation Learning with Deep Convolutional Generative Adversarial Networks by Alec Radford, Luke Metz, Soumith Chintala.

The generator and discriminator are all convolutional networks without max pooling but instead uses strided (and fractionally strided) convolutions. Fruthermore, batch normalization is used and ReLu and Leaky ReLU activations.

Usage

Usage: dcgan-mnist.py [OPTIONS]

Options:
  --root TEXT              Root directory for MNIST dataset
  --epochs INTEGER         Number of epochs
  --batch-size INTEGER     Batch size
  --latent-vector INTEGER  Size of latent vector Z
  --disable-cuda TEXT      Disable CUDA acceleration
  --help                   Show this message and exit.

License

dcgan-mnist is Copyright © 2019 Alexander Stante. It is free software, and may be redistributed under the terms specified in the LICENSE file.

About

Deep Convolutional Generative Adversarial Network for MNIST Dataset

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages