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

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

object-detection-api object-detection tensorflow c++

2

Hossein · 技术社区 · 8 年前

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

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

2 回复 | 直到 8 年前

1

6

Mario García 7 年前

我认为最简单的方法是使用 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

gdelab 8 年前

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

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