TensorFlow，如何保存样式迁移模型以备后用？

Question

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

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

Answer 1

使用model.save() 方法。

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