在Java中加载Tensorflow模型

在Java中加载Tensorflow模型,java,tensorflow,Java,Tensorflow,我尝试使用我训练过的tensorflow模型在Java中进行对象检测。 我用这个脚本导出了模型(用Python训练)。这将生成一个冻结的\u推断\u graph.pb文件以及一个保存的\u model.pb文件 我的理解是,在Java中使用此模型的最佳方法是通过SavedModelBundle.load()加载模型。但是,当我在保存的_model.pb文件中尝试此操作时,会出现错误SavedModel未在导出目录中找到:。有没有更好的方法在Java中加载模型以进行对象检测 如果你有任何建议,请告

我尝试使用我训练过的tensorflow模型在Java中进行对象检测。 我用这个脚本导出了模型(用Python训练)。这将生成一个冻结的\u推断\u graph.pb文件以及一个保存的\u model.pb文件

我的理解是,在Java中使用此模型的最佳方法是通过SavedModelBundle.load()加载模型。但是,当我在保存的_model.pb文件中尝试此操作时,会出现错误
SavedModel未在导出目录中找到:
。有没有更好的方法在Java中加载模型以进行对象检测


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

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

你可能会发现这本书也很有启发性


希望有帮助。

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

你可能会发现这本书也很有启发性

希望有帮助