全部完成时代码结构如下:
使用MS COCO数据集:
放到train2014下。
该数据集解压后18G左右,共328000张图片,每张style-image训练的话,在服务器GPU环境下需要4-5个小时左右,迭代次数50000多次,最终Loss收敛在六七十。
测试使用模型:
扫描二维码关注公众号,回复:
1733281 查看本文章
可以看到,使用cubist风格转换后还不错
最终训练结果:
全部完成时代码结构如下:
使用MS COCO数据集:
放到train2014下。
该数据集解压后18G左右,共328000张图片,每张style-image训练的话,在服务器GPU环境下需要4-5个小时左右,迭代次数50000多次,最终Loss收敛在六七十。
测试使用模型:
可以看到,使用cubist风格转换后还不错
最终训练结果: