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

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/qq_27855219/article/details/89371186

按照各位大佬博客,我也利用萌妹子头像训练了一下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:

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

猜你喜欢

转载自blog.csdn.net/qq_27855219/article/details/89371186