访问官网
Github使用TensorFlow实现MNIST数据集的GAN
GitHub - yihui-he/GAN-MNIST: 使用TensorFlow实现MNIST数据集的生成对抗网络
TensorFlow实现
- 本项目中的所有代码都是Theano版本的简单复制
代码
- 位于
face/和mnist/目录下 - model.py
- DCGAN模型的定义
- train.py
- 训练DCGAN模型（并不时生成样本）
- util.py
- 图像相关的工具函数
数据集
- MNIST
- CelebA人脸数据集
- http://mmlab.ie.cuhk.edu.hk/projects/CelebA.html
- 下载"img_align_celeba"图像
- 根据下载的数据集路径，在train.py中设置"face_image_path"
参考资料
https://github.com/carpedm20/DCGAN-tensorflow
引用
如果您在研究中发现本代码有用，请考虑引用：
@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}
}
