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 模型:

  1. 加载模型:
TensorFlowInferenceInterface inferenceInterface = 
     new TensorFlowInferenceInterface(assetManager, modelFilename);
  1. 将输入数据发送到 TensorFlow 二进制文件:
inferenceInterface.feed(inputName, 
    floatValues, 1, inputSize, inputSize, 3);
  1. 运行预测或推理:
inferenceInterface.run(outputNames, logStats);
  1. 接收 TensorFlow 二进制文件的输出:
inferenceInterface.fetch(outputName, outputs);

results matching ""

    No results matching ""