深度学习DCGAN自己的数据集实现

按照各位大佬博客，我也利用萌妹子头像训练了一下DCGAN。本人电脑CPU的 4GB 啊啊啊，100epoch跑了一天。不过看到最后生成了清晰的萌妹子头像还是很开心的。。。。。

我github上的这篇文章的代码地址：

原DCGAN的github地址：https://github.com/carpedm20/DCGAN-tensorflow

DCGAN

1.网络结构图：

G网络：

D网络： 

2.代码实现：

trick：

(1)防止d_loss为0，G训练2次，D训练一次。

(2)BatchNormalizer加快收敛速度。

(3)D网络Relu改为leakyrelu（0.2）。

(3)将输入图像的rgb[0,255]归一化到[-1,1]。

数据集：某位大佬抓取并截取好的二次元萌妹子人脸。

文件:

main.py: 主程序，主要是参数配置

model.py: G、D网络模型及训练过程

utils.py: 图像处理及可视化。

ops.py: 定义运算，例如卷积、反卷积、leakyrelu、BN层、linear层

download.py: 下载数据集

基于DCGAN源码，主要修改的位置：

input_height: 改成数据集图片的尺寸，这里我的是96

out_height: 随便改，但最好不要大于原图尺寸，这里我改成64

dataset：这是放数据集的文件名

data_dir：放数据集的文件的目录

train、crop、visualize：true

所有的代码参考我的github地址:

实验结果：

1epoch：

50epoch：

100epoch：

125epoch：

训练时间长了可能会更好一点，奈何我的破电脑跑太慢了。。。。。