来自 TensorFlow 的用于图像分类的迁移学习 API

Question

我一直在检查来自 tensorflow 的最新版本的 retrain.py，但我在训练的图形和保存的模型之间遇到了混淆。详细说明：

1. 在retrain.py 的第 1307 行中，我们给出了保存“训练图”的路径。这将保存为 .pb 文件。然后我们使用这个图来进一步分类我们的数据集（使用label_image.py）

2. 在新版retrain.py中，还有一个saved_model路径（在1482行）。这也保存了一个 .pb 文件。代码说明它是“服务”的“导出模型”。

我不明白代码生成的两个 .pb 文件（output_graph.pb 和 saved_model.pb）之间的区别。 saved_model.pb 的重要性是什么，可以在哪里使用？这里的“服务”是什么意思？

任何解释或可能的解释链接都会有很大帮助。

感谢 N 问候，

普拉迪普

Answer 1

Tensorflow Serving 是一个服务系统，可用于使用 docker 容器使您的模型在生产中可用。新格式以可用作服务对象的方式保存模型。