0
点赞
收藏
分享

微信扫一扫

用DGAN生成mnist图片以及动漫头像图片

生态人 2022-01-15 阅读 74

第一次接触GAN的时候就觉得这个东西十分的妙~
生成mnist图片跟生成动漫头像的代码大同小异,那么接下来就主要讲生成DGAN生成动漫头像。

一 . 数据集

数据集链接:
链接:https://pan.baidu.com/s/1evXwE7DmAJ0wKXw96Ojj_w
提取码:liao
在这里插入图片描述
关于数据处理,因为一开始网上找的图片的名字太长,不太方便后面后续做数据集,就统一修改了一下图片名字。
然后就是自己定义自己的数据集

  1. 生成txt文件,里面包含图片路径,以及对应的标签,由于GAN是无监督的,所以我就把全部图片标签都设置为1
    在这里插入图片描述

  2. 然后就是定义自己的数据集啦
    在这里插入图片描述
    在这里插入图片描述

二. 网络结构设计

在这里插入图片描述

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
训练代码
在这里插入图片描述

三. 结果

只跑了22次,跑不动了/(ㄒoㄒ)/~~,多跑几次效果应该会更好吧
在这里插入图片描述
关于为什么生成图片是23个,并且拼接在一起?
因为调用的save_image函数会调用make_grid函数进行拼接,一行默认8个
之所以是23个,是因为每一次迭代,都会随机取128(这里我设置的是128)个小批数据,而一共有51223个数据,这里取的最后一批,51223%128的余数是23,因此是23个图片。

全代码链接
https://gitee.com/LiaoTing2333/GAN.git

举报

相关推荐

0 条评论