Github|教你用PyTorch打造专属GAN二次元头像(附源代码)

Posted 2021-05-02 全球人工智能

全球人工智能

---

来源：Github Jie Le

翻译：孙菁

随机生成图像

通过DCGAN模型在包含143,000张动漫人脸的数据集上迭代训练100次，我们得到了本次实验所用的图像。

图像转换

通过执行隐含代码，使图像完成从第一列到最后一列的转换。

原始图像

我们可以观察到图像包含很多异常值，即有很多噪音，这些异常值会使得生成图像的质量有所下降。

使用

执行以下命令以启动整个程序：

$ python main.py --dataRoot path_to_dataset/ 

DCGAN的预训练模型呀包含在此目录中，使用jupyternotebook工具即可查看。

动漫人脸数据集

利用爬虫工具 gallery-dl，我们 从 danbooru.donmai.us中收集到了带有126个标记的动漫样式图片。之后，这些图片由动漫人脸识别器 python-animeface 处理。经过处理的数据集总共包含143,000张动漫人脸。需要注意的是，一些标记在经过剪裁后就没有意义了，比如，一张在“制服”标签下的图片在剪裁后就不再包含制服的部分了。

如何从头构件数据集？

预备工具：gallery-dl, python-animeface

1、下载动漫样式的图片

2、从下载的图片中提取动漫人脸

读者可以从build_face_dataset.py中寻找具体实现。

数据集（~400MB）也可以从此处下载（非商业使用）。

Github库点击阅读原文跳转

热门文章推荐

重磅|MIT发布脑控机器人:用脑电波（10毫秒分类）纠正机器人错误