代码之家  ›  专栏  ›  技术社区  ›  Hossein

Tensorflow对象检测API有C++包装器吗?

  •  2
  • Hossein  · 技术社区  · 7 年前

    我们已经训练了我们的模型,并使用提供的Python脚本成功地测试了它们。然而,我们现在想在我们的网站上部署它,并为第二轮测试运行一个web服务。

    是否有一个C++包装器,以便我们可以用与Python脚本相同的方式运行/执行我们的模型?

    2 回复  |  直到 7 年前
        1
  •  6
  •   Mario García    5 年前

    我认为最简单的方法是使用 cppflow . 它是TensorFlow C API的C++包装器。它很简单,但真正易于使用,您不需要安装它,也不需要使用Bazel编译。您只需下载C API并按如下方式使用:

    Model model("graph.pb");
    model.restore("path/to/checkpoint");
    
    auto input = new Tensor(model, "input");
    auto output = new Tensor(model, "output");
    
    model.run(input, output);
    
        2
  •  2
  •   gdelab    7 年前

    您将找到在C上运行对象检测的代码++ here . 您需要一个导出的图形(.pb格式),可以使用 the TF object detection API .

    编译过去很复杂(除非您将项目放在tensorflow目录中,并使用bazel编译所有内容,但您可能不想这样做)。我认为现在应该更容易了,但我不知道怎么做;或者你可以跟随 these instructions 自行编译tensorflow并在cmake项目中使用。还有一个用c运行图的例子++ here .