来自Tensorflow的图像分类转移学习API

时间:2018-04-13 04:04:57

标签: python tensorflow

我一直在研究tensorflow的retrain.py的最新版本,但是我在训练有素的图表和保存的模型之间遇到了困惑。详细说明:

  1. retrain.py的第1307行中,我们提供了保存“训练图”的路径。这将保存为.pb文件。然后,我们使用此图表对我们的数据集进行进一步分类(使用label_image.py

  2. retrain.py的新版本中,还有一个saved_model路径(在第1482行)。这也会保存.pb文件。该代码解释说它是“服务”的“导出模型”。

  3. 我不明白代码生成的两个.pb文件( output_graph.pb saved_model.pb )之间的区别。 saved_model.pb的重要性和使用位置是什么? “服务”在这里意味着什么?

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

    谢谢N问候,

    普拉迪普

1 个答案:

答案 0 :(得分:1)

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