按照各位大佬博客,我也利用萌妹子头像训练了一下DCGAN。本人电脑CPU的 4GB 啊啊啊,100epoch跑了一天。不过看到最后生成了清晰的萌妹子头像还是很开心的。。。。。
我github上的这篇文章的代码地址:https://github.com/Rui012345/tensorflowCNN/tree/master/DCGAN-mydataset
原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)将输