GAN网络之入门教程(四)之基于DCGAN动漫头像生成
目录 使用前准备 数据集 定义参数 构建网络 构建G网络 构建D网络 构建GAN网络 关于GAN的小trick 训练 总结 参考 这一篇博客以代码为主,主要是来介绍如果使用keras构建一个DCGAN,然后基于DCGAN,做一个自动生成动漫头像。训练过程如下(50轮的训练过程)“ 关于DCGAN或者GAN的相关知识,可以参考 GAN网络入门教程 。建议先了解相关知识,再来看这一篇博客。 项目地址: GitHub 使用前准备 首先的首先,我们肯定是需要数据集的,这里使用的数据集来自 kaggle——Anime Faces 。里面有21551张动漫头像的图片。大家可以到kaggle上面去下载数据集,或者说到我的 github 上去下载数据集(求个 ⭐ 不过分吧)。部分数据如下: 如果自己电脑计算机资源不是很强的话,比如我,一个mx250小水管(玩玩lol还是可以的,训练这个模型可能要等到下辈子),推荐大家去注册一个kaggle或者colab账号去白嫖GPU资源(1080,2080的玩家请随意)。不过个人更加的推荐kaggle,因为感觉它的资源分配是可见的,且可以后台运行。 数据集 数据集是动漫图片,我们可以将图片的像素点的值变成 \([-1,1]\) 之间,具体代码如下: # 数据集的位置 avatar_img_path = "./data" import imageio import