背景
打算利用AnimGANv2训练一个动漫风格的模型,不过结果并不是很成功,AI似乎误会了什么,可以看到头发有大片的白色。先简单记录一下训练的过程吧
下载AnimGANv2项目
准备数据
AnimGANv2训练的化需要规范成256x256的真实图片和规范成256x256的动漫图片,真实图片我是从flickr批量扒下来的,通过YunNet抽取脸部图片,规范成256 X 256
可以把之前的dataset/train_photo
重命名,然后再建立一个train_photo
目录,将数据放到dataset/train_photo
目录下
动漫图片我是从网上下载的一个图片集,原本是512x512,也规范化成256x256
在dataset目录下建立Custom/style
目录,将动漫图片都放在里面,调用下面的方法自动生成训练需要的smooth文件夹
python edge_smooth.py --dataset Custom --img_size 256
训练
训练直接使用下列脚本即可
python train.py --dataset Custom --epoch 101 --init_epoch 10
这里的Custom就是对应上面建立的动漫目录文件夹名字
失败原因分析
训练的动漫头像背景都是纯白色的,不知道是否会被误认为成一种风格,可以尝试从动漫视频中截取一些带头像的图片再次进行训练看看效果