在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也具有指导意义。

希望有所帮助。