在 Java 中加载 TensorFlow 模型

Question

我正在尝试使用我训练有素的 tensorflow 模型在 Java 中进行对象检测。 我使用此脚本export_inference_graph 导出了模型（使用 Python 训练）。这会生成一个 frozen_inference_graph.pb 文件以及一个 saved_model.pb 文件。

我的理解是，在 Java 中使用这个模型的最好方法是通过 SavedModelBundle.load() 来加载模型。但是，当我使用 saved_model.pb 文件尝试此操作时，我收到错误 SavedModel not found in export directory:。有没有更好的方法在 Java 中加载模型以进行对象检测？

如果您有任何建议，请告诉我！我怀疑我在这个过程中做错了什么。

Answer 1

SavedModel 格式将所有模型信息编码在一个目录中，而不是一个文件中。因此，您希望将包含saved_model.pb 文件的目录提供给SavedModelBundle.load()，而不是文件本身。

您可能会发现official sample 也很有指导意义。

希望对您有所帮助。