/dcgan-pytorch

Generating fake images using DCGAN (Deep convolutional generative adversarial networks) and small NORB dataset

Primary LanguagePythonApache License 2.0Apache-2.0

Requirements

pytorch >= 0.4.1

Setup Environment

# Dependencies
pip install matplotlib
pip install git+https://github.com/wookayin/tensorflow-plot.git@master

Dataset

Download small NORB dataset

cd scripts
./download_norb_small.sh

Train

./run.sh

Results

alt text alt text

Acknowledgment

https://github.com/ndrplz/small_norb