是否有适用于张量流对象检测 API 的C++包装器

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

本文关键字：C++ 包装 API 检测 适用于 张量流 对象 是否      更新时间：2023-10-16

我们已经训练了我们的模型，并使用提供的Python脚本成功地测试了它们。但是，我们现在希望将其部署在我们的网站上，并为第二轮测试运行Web服务。

是否有一个C++包装器，以便我们可以像使用 Python 脚本一样运行/执行我们的模型？

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

在此处找到在C++上运行对象检测的代码。你需要一个导出的图形（.pb 格式），可以使用 TF 对象检测 API 获取该图形。

编译曾经很棘手（除非你把你的项目放在 tensorflow 目录中并使用 bazel 编译所有内容，但你可能不想这样做）。我认为现在应该更容易，但我不知道怎么做;或者你可以按照这些说明自行编译 TensorFlow，并在 CMAKE 项目中使用它。这里有另一个在 c++ 中运行图形的示例。

相关文章：