李宏毅机器学习GAN作业之动漫头像生成记录

数据集

    36740张3*64*64大小的二次元人物头像:

李宏毅机器学习GAN作业之动漫头像生成记录_第1张图片

 模型

本次采用DCGAN实现

论文名称:Unsupervised Representation Learning with Deep Convolutional Generative Adversarial Networks

论文地址:https://arxiv.org/abs/1511.0643

实验参数配置

学习率:

G:2e-4

D:2e-4

batch_size = 64

D每个step都进行训练,G每五个step进行一次训练。

实验结果

训练20epoch

李宏毅机器学习GAN作业之动漫头像生成记录_第2张图片

 训练100epoch

李宏毅机器学习GAN作业之动漫头像生成记录_第3张图片

训练500epoch

 李宏毅机器学习GAN作业之动漫头像生成记录_第4张图片

 发现epoch200之后变化并不是很大,可能是学习率太大了,但是太菜了,又不会动态学习率,然后改变学习率,加载500epoch的权重继续训练。

学习率:

G:   1e-5

D:1e-5

训练结果:

训练100epoch:

李宏毅机器学习GAN作业之动漫头像生成记录_第5张图片

提升了一点,后续继续改进。

现在属于小白水平,不断学习不断进步,后续继续补充。

你可能感兴趣的:(生成对抗网络,人工智能,神经网络)