全部完成时代码结构如下：
使用MS COCO数据集：
放到train2014下。
该数据集解压后18G左右，共328000张图片，每张style-image训练的话，在服务器GPU环境下需要4-5个小时左右，迭代次数50000多次，最终Loss收敛在六七十。
测试使用模型：
可以看到，使用cubist风格转换后还不错
最终训练结果：
【创新实训】风格迁移功能探索与实现（六）完成代码并训练 保存训练结果 测试效果
猜你喜欢
转载自blog.csdn.net/qq_33837704/article/details/80658509
今日推荐
周排行