GAN on MNIST with TensorFlow

GitHub - yihui-he/GAN-MNIST: Generative Adversarial Network for MNIST with tensorflow

You can't perform that action at this time. You signed in with another tab or window. You signed out in another tab or window. Reload to refresh your session. Reload to refresh your session.

https://github.com/yihui-he/GAN-MNIST

Tensorflow implementation

• All the codes in this project are mere replication of Theano version

Code

• Under face/ and mnist/

• model.py

• Definition of DCGAN model

• train.py

• Training the DCGAN model (and Generating samples time to time)

• util.py

• Image related utils

Dataset

• MNIST

• http://yann.lecun.com/exdb/mnist/

• CelebA Face dataset

• http://mmlab.ie.cuhk.edu.hk/projects/CelebA.html

• Download “img_align_celeba” images

• Set “face_image_path” in train.py according to the path of downloaded dataset

references

Citation

@InProceedings{He_2017_ICCV,
author = {He, Yihui and Zhang, Xiangyu and Sun, Jian},
title = {Channel Pruning for Accelerating Very Deep Neural Networks},
booktitle = {The IEEE International Conference on Computer Vision (ICCV)},
month = {Oct},
year = {2017}
}