Anim_GAN:PyTorch Gan自动生成的动漫头像 源码
2021.03.04 最近看了一些有关GAN对抗神经网络的书,然后看的一些书上有扭曲GAN生成各种图像的程序,自己也尝试着跟着pytorch的工具书写了一个非常简陋的GAN对抗生成动漫头像的小项目 训练集(动漫头像主要来自于safe.booru.donmai.us),可以非常方便的通过爬虫进行爬取,因为爬取的网页一页有20张jpg,所以计数单位是页码运行download.py或download_threads即可,通常来说1000个数据集有点过少了,可以适当增加到10000到100000个 下载完训练集后直接运行main.py即可开始训练,checkpoints中是我的预训练放置的模型,因为训练的epoch不是很多,效果只有轮廓。
文件列表
Anim_GAN-main.zip
(预估有个13文件)
Anim_GAN-main
config.py
538B
download.py
2KB
checkpoints
README.md
41B
imgs
README.md
21B
Myutils.py
634B
main.py
3KB
model.py
2KB
暂无评论