Tensorflow已提供此tutorial作为在Android应用程序上使用tensorflow的示例。

我已经成功构建了该应用程序,并在我的设备上启动了该应用程序。我也阅读了代码。

但是我不知道从这里去哪里!我有一个Tensorflow代码,可以根据给定的训练数据来计算准确性。

如何在自己的android应用中使用该模型。 tensorflow描述不超出构建说明。

请帮忙!

最佳答案

首先,您需要将模型保存到.pb(protobuf)文件。
在您的应用程序中,您将需要加载此.pb文件。您可以使用tf/python/tools freeze_graph.freeze_graph()实现

您还需要libtensorflow_inference.so,其中包含所有c ++实现。

最后,您将需要libandroid_tensorflow_inference_java.jar。

使用这些,您可以编写:

inferenceInterface.initializeTensorFlow(getAssets(), MODEL_FILE);

inferenceInterface.fillNodeFloat(INPUT_NODE, INPUT_SIZE, inputFloats);

inferenceInterface.runInference(new String[] {OUTPUT_NODE});

inferenceInterface.readNodeFloat(OUTPUT_NODE, resu);


这里有一个很棒的教程:https://omid.al/posts/2017-02-20-Tutorial-Build-Your-First-Tensorflow-Android-App.html

该应用程序的GitHub存储库:https://github.com/omimo/TFDroid

对我而言,这对理解Android上tf的基础很有帮助。

09-27 05:32