在c ++ app中使用Tensorflow模型

时间:2016-04-10 16:08:27

标签: android python c++ tensorflow tensorflow-serving

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

今天我花了一整天的时间来完成Tensorflow服务教程:https://tensorflow.github.io/serving/serving_basic

我已安装并正常工作,但对我来说,下一步该做什么仍然不明显。 TF服务是正确的吗?它可以用于将训练有素的TF模型集成到应用程序中吗?或者只是构建可以从终端运行的模型?

如果TF服务不是我需要的,那么我应该在哪里寻找呢?是否有更简单的方法来使用经过训练的跨平台模型?

谢谢!

1 个答案:

答案 0 :(得分:0)

请在教程中查看此section,了解如何在c ++中加载SessionBundle

拥有SessionBundle后,您可以使用signature.h中的实用程序来获取签名(例如GetClassificationSignature),然后运行推理(例如RunClassification)。