代码之家  ›  专栏  ›  技术社区  ›  David Papp

在Java中加载Tensorflow模型

  •  0
  • David Papp  · 技术社区  · 7 年前

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

    我的理解是,在Java中使用此模型的最佳方法是通过SavedModelBundle加载模型。加载()。但是,当我用保存的\ U模型尝试此操作时。pb文件,我得到错误 SavedModel not found in export directory: . 有没有更好的方法在Java中加载模型以进行对象检测?

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

    1 回复  |  直到 7 年前
        1
  •  2
  •   ash    7 年前

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

    您可以找到 official sample 也要有教育意义。

    希望这有帮助。