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的基础很有帮助。