python - TenserFlow，如何保存样式传输模型供以后使用？

Question

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

如何保存训练好的模型？保存后如何在另一张内容照片上使用它？

标签： pythontensorflowstyle-transfer

Answer 1

使用model.save()方法。

阅读本教程：https ://www.tensorflow.org/guide/keras/save_and_serialize