记录一次并不成功的AnimGANv2模型训练

355 阅读1分钟

背景

打算利用AnimGANv2训练一个动漫风格的模型,不过结果并不是很成功,AI似乎误会了什么,可以看到头发有大片的白色。先简单记录一下训练的过程吧

image.png

下载AnimGANv2项目

GitHub - TachibanaYoshino/AnimeGANv2: [Open Source]. The improved version of AnimeGAN. Landscape photos/videos to anime下载项目,配置好相关依赖

准备数据

AnimGANv2训练的化需要规范成256x256的真实图片和规范成256x256的动漫图片,真实图片我是从flickr批量扒下来的,通过YunNet抽取脸部图片,规范成256 X 256

image.png

可以把之前的dataset/train_photo重命名,然后再建立一个train_photo目录,将数据放到dataset/train_photo目录下

动漫图片我是从网上下载的一个图片集,原本是512x512,也规范化成256x256

image.png

在dataset目录下建立Custom/style目录,将动漫图片都放在里面,调用下面的方法自动生成训练需要的smooth文件夹

python edge_smooth.py --dataset Custom --img_size 256

训练

训练直接使用下列脚本即可

python train.py --dataset Custom --epoch 101 --init_epoch 10

这里的Custom就是对应上面建立的动漫目录文件夹名字

失败原因分析

训练的动漫头像背景都是纯白色的,不知道是否会被误认为成一种风格,可以尝试从动漫视频中截取一些带头像的图片再次进行训练看看效果