【创新实训】风格迁移功能探索与实现(六)完成代码并训练 保存训练结果 测试效果

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