我一直在使用TensorFlow 网站上的本教程来创建执行样式转换的脚本(python3)。我正在尝试在特定艺术作品上训练模型,然后将该风格应用于任何随机照片。根据我对本教程的理解,该脚本采用样式图像和内容图像,通过 VGG19 模型运行它们并输出最终图像(在我的机器上大约需要 30 分钟)。但我看不出有办法保存训练后的模型以将其应用于另一张内容照片。本教程没有像我期望的那样使用 TF 的模型fit()
、predict()
和方法。save()
它似乎只是在训练时将预测应用于图像。
如何保存训练好的模型?保存后如何在另一张内容照片上使用它?