2021.03.04
最近看了一些有关GAN对抗神经网络的书,然后看的一些书上有扭曲GAN生成各种图像的程序,自己也尝试着跟着pytorch的工具书写了一个非常简陋的GAN对抗生成动漫头像的小项目
训练集(动漫头像主要来自于safe.booru.donmai.us),可以非常方便的通过爬虫进行爬取,因为爬取的网页一页有20张jpg,所以计数单位是页码运行download.py或download_threads即可,通常来说1000个数据集有点过少了,可以适当增加到10000到100000个
下载完训练集后直接运行main.py即可开始训练,checkpoints中是我的预训练放置的模型,因为训练的epoch不是很多,效果只有轮廓。
2021-11-24 11:38:09
26KB
Python
1