第一次接触GAN的时候就觉得这个东西十分的妙~
生成mnist图片跟生成动漫头像的代码大同小异,那么接下来就主要讲生成DGAN生成动漫头像。
一 . 数据集
数据集链接:
链接:https://pan.baidu.com/s/1evXwE7DmAJ0wKXw96Ojj_w
提取码:liao
关于数据处理,因为一开始网上找的图片的名字太长,不太方便后面后续做数据集,就统一修改了一下图片名字。
然后就是自己定义自己的数据集
-
生成txt文件,里面包含图片路径,以及对应的标签,由于GAN是无监督的,所以我就把全部图片标签都设置为1
-
然后就是定义自己的数据集啦
二. 网络结构设计
训练代码
三. 结果
只跑了22次,跑不动了/(ㄒoㄒ)/~~,多跑几次效果应该会更好吧
关于为什么生成图片是23个,并且拼接在一起?
因为调用的save_image函数会调用make_grid函数进行拼接,一行默认8个
之所以是23个,是因为每一次迭代,都会随机取128(这里我设置的是128)个小批数据,而一共有51223个数据,这里取的最后一批,51223%128的余数是23,因此是23个图片。