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

Dear 丶 2022-04-24 06:06 699阅读 0赞

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

我github上的这篇文章的代码地址：[https://github.com/Rui012345/tensorflowCNN/tree/master/DCGAN-mydataset][https_github.com_Rui012345_tensorflowCNN_tree_master_DCGAN-mydataset]

原DCGAN的github地址：[https://github.com/carpedm20/DCGAN-tensorflow][https_github.com_carpedm20_DCGAN-tensorflow]

# **DCGAN** #

## **1.网络结构图：** ##

### **G网络：** ###

## ![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzI3ODU1MjE5_size_16_color_FFFFFF_t_70][] ##

### D网络：  ###

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzI3ODU1MjE5_size_16_color_FFFFFF_t_70 1][]

## 2.代码实现： ##

trick：

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

(2)BatchNormalizer加快收敛速度。

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

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

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

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzI3ODU1MjE5_size_16_color_FFFFFF_t_70 2][]

文件:

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

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzI3ODU1MjE5_size_16_color_FFFFFF_t_70 3][]

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

实验结果：

1epoch：

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzI3ODU1MjE5_size_16_color_FFFFFF_t_70 4][]

50epoch：

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzI3ODU1MjE5_size_16_color_FFFFFF_t_70 5][]

100epoch：

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzI3ODU1MjE5_size_16_color_FFFFFF_t_70 6][]

125epoch：

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzI3ODU1MjE5_size_16_color_FFFFFF_t_70 7][]

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

[https_github.com_Rui012345_tensorflowCNN_tree_master_DCGAN-mydataset]: https://github.com/Rui012345/tensorflowCNN/tree/master/DCGAN-mydataset
[https_github.com_carpedm20_DCGAN-tensorflow]: https://github.com/carpedm20/DCGAN-tensorflow
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzI3ODU1MjE5_size_16_color_FFFFFF_t_70]: /images/20220218/739b82f472024a12accfa0de31345507.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzI3ODU1MjE5_size_16_color_FFFFFF_t_70 1]: /images/20220218/27de40ffc6d841fca463b63456c10d60.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzI3ODU1MjE5_size_16_color_FFFFFF_t_70 2]: /images/20220218/2f140a2e0a334468959489f345e04feb.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzI3ODU1MjE5_size_16_color_FFFFFF_t_70 3]: /images/20220218/20024c6bee624dfcb71157a33f3adccc.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzI3ODU1MjE5_size_16_color_FFFFFF_t_70 4]: /images/20220218/35f4384be39c440e876b1ef4397f399d.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzI3ODU1MjE5_size_16_color_FFFFFF_t_70 5]: /images/20220218/9f77d19bcfbf4892902924d9e973a89d.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzI3ODU1MjE5_size_16_color_FFFFFF_t_70 6]: /images/20220218/9fb003739152449a96ce59b95196f8eb.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzI3ODU1MjE5_size_16_color_FFFFFF_t_70 7]: /images/20220218/376b653818624c9498f989dd6dcb7e34.png