在Java中加载Tensorflow模型

1 周，1 日 Questions & Answers 4808

我试图用我训练过的tensorflow模型在Java中进行对象检测。我用这个脚本export_inference_graph导出了模型（用Python训练）。这生成了一个冻结的推理图。pb文件以及保存的_模型。pb文件

我的理解是，在Java中使用该模型的最佳方式是通过SavedModelBundle加载该模型。加载（）。然而，当我尝试使用保存的_模型时。pb文件，我得到错误SavedModel not found in export directory:。有没有更好的方法在Java中加载模型以进行对象检测