是否有用于 Tensorflow 对象检测的 C++ 包装器 API?

Is there a C++ wrapper for the Tensorflow Object Detection API?

我们已经使用提供的 Python 脚本训练并成功测试了我们的模型。但是,我们现在想将其部署在我们的网站上,并 运行 用于第二轮测试的网络服务。

是否有 C++ 包装器,以便我们可以像使用 Python 脚本一样使用 run/execute 我们的模型?

您将在 C++ here. You'll need an exported graph (.pb format), that you can get using the TF object detection API 上找到 运行 对象检测的代码。

编译曾经很棘手(除非你将项目放在 tensorflow 目录中并使用 bazel 编译所有内容,但你可能不想这样做)。我想现在应该更容易了,但我不知道怎么做;或者你可以关注 these instructions to compile tensorflow on its own and use it in a cmake project. You have another example of runing a graph in c++ .

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