Android 应用中的 TF Mobile
TensorFlow 生态系统使其能够通过接口类TensorFlowInferenceInterface
,和 jar 文件libandroid_tensorflow_inference_java.jar
中的 TensorFlow Java API 在 Android 应用中使用。您可以使用 JCenter 中的 jar 文件,从ci.tensorflow.org
下载预编译的 jar,也可以自己构建。
推理界面已作为 JCenter 包提供,可以通过将以下代码添加到build.gradle
文件中包含在 Android 项目中:
allprojects {
repositories {
jcenter()
}
}
dependencies {
compile 'org.tensorflow:tensorflow-android:+'
}
您可以按照此链接中的说明使用 Bazel 或 Cmake 自行构建它们,而不是使用 JCenter 中的预构建二进制文件:https://github.com/tensorflow/tensorflow/blob/r1.4/tensorflow/contrib/android/README.md.
在 Android 项目中配置 TF 库后,您可以通过以下四个步骤调用 TF 模型:
- 加载模型:
TensorFlowInferenceInterface inferenceInterface =
new TensorFlowInferenceInterface(assetManager, modelFilename);
- 将输入数据发送到 TensorFlow 二进制文件:
inferenceInterface.feed(inputName,
floatValues, 1, inputSize, inputSize, 3);
- 运行预测或推理:
inferenceInterface.run(outputNames, logStats);
- 接收 TensorFlow 二进制文件的输出:
inferenceInterface.fetch(outputName, outputs);