使用自定义 TensorFlow Lite 版本

如果您是一位经验丰富的机器学习开发者,并且预构建的 TensorFlow Lite 库不能满足您的需求,您可以通过机器学习套件使用自定义 TensorFlow Lite build。例如,您可能需要添加自定义操作。

前提条件

捆绑适用于 Android 的自定义 TensorFlow Lite

构建 Tensorflow Lite AAR:

bazel build --cxxopt='--std=c++11' -c opt        \
  --fat_apk_cpu=x86,x86_64,arm64-v8a,armeabi-v7a   \
  //tensorflow/lite/java:tensorflow-lite

这将在 bazel-genfiles/tensorflow/lite/java/ 中生成一个 AAR 文件。将自定义的 Tensorflow Lite AAR 发布到您的本地 Maven 代码库:

mvn install:install-file -Dfile=bazel-genfiles/tensorflow/lite/java/tensorflow-lite.aar -DgroupId=org.tensorflow \
  -DartifactId=tensorflow-lite -Dversion=0.1.100 -Dpackaging=aar

最后,在您的应用 build.gradle 中,使用您的自定义版本替换 Tensorflow Lite:

implementation 'org.tensorflow:tensorflow-lite:0.1.100'