我正在尝试使用我训练过的tensorflow模型在Java中进行对象检测。 我用这个脚本导出了模型(用Python训练) export_inference_graph . 这生成了冻结的\u推理图。pb文件以及保存的\u模型。pb文件。
我的理解是,在Java中使用此模型的最佳方法是通过SavedModelBundle加载模型。加载()。但是,当我用保存的\ U模型尝试此操作时。pb文件,我得到错误 SavedModel not found in export directory: . 有没有更好的方法在Java中加载模型以进行对象检测?
SavedModel not found in export directory:
如果你有任何建议,请告诉我!我怀疑我在这个过程中做错了什么。
SavedModel格式编码目录中的所有模型信息,而不是文件。因此,您希望提供包含 saved_model.pb 文件收件人 SavedModelBundle.load() ,而不是文件本身。
saved_model.pb
SavedModelBundle.load()
您可以找到 official sample 也要有教育意义。
希望这有帮助。