在 C++ 应用程序中使用 TensorFlow 模型

Question

提前为这个相对新手的问题道歉。使用 Tensorflow，我在 python 中训练了一个神经网络，我想用它来对 c++ 应用程序中的图像进行分类，我想稍后将其集成到一个 android 应用程序中。

今天我花了一整天的时间学习 Tensorflow Serving 教程：https://tensorflow.github.io/serving/serving_basic。

我已经安装好所有东西，并且可以正常工作，但我仍然不清楚下一步该做什么。 TF Serving 是正确的做法吗？它可以用于将经过训练的 TF 模型集成到应用程序中吗？还是只是构建可以从终端运行的模型？

如果 TF Serving 不是我所需要的，那么我应该去哪里寻找呢？有没有更简单的方法可以跨平台使用经过训练的模型？

谢谢！

Answer 1

请查看此section在教程中如何在C ++中加载SessionBundle。

一旦您有一个会话栏，您可以使用signature.h的实用程序获取签名（例如，GetClassificationSignature）然后运行推理（例如，RunClassification）。