使用StyleGAN2生成动漫头像(二)

经过四天的等待,单卡Tesla训练4000kimg终于完成,最终模型有300M大小。但是生成的图像中我们发现一些面部特征扭曲,存在严重缺陷的头像,大概是一开始没有清理完数据所致,这时候我们用CNN做一个筛选。

将上面的图片打上Bad标签,剩下的正常图片打上Good标签,训练即可,最终的accuracy在92%左右,因为只是大概筛选,这时候已经可以满足要求了。

上图就是最后的成品啦~

  • 版权声明: 本博客所有文章除特别声明外,均采用 CC BY-NC-SA 许可协议。转载需要标明作者,并以超链接的方式指向原文。
  • Copyrights © 2020 Kevin Li
  • 访问人数: | 浏览次数:

请我喝杯咖啡吧~