在c++应用程序中使用Tensorflow模型

Use Tensorflow model in c++ app

本文关键字：Tensorflow 模型 c++ 应用程序更新时间：2023-10-16

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

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

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

如果TF服务不是我需要的，那么我应该在哪里寻找替代？有没有一种更简单的方法可以跨平台使用经过训练的模型？

谢谢！

请参阅教程中的这一节，了解如何在c++中加载SessionBundle。

一旦您有了SessionBundle，您就可以使用signature.h中的实用程序来获得签名（例如，GetClassificationSignature），然后运行推理（例如，RunClassification）。