使用自定義 TensorFlow Lite 構建

如果您是經驗豐富的 ML 開發人員,且預先建置的 TensorFlow Lite 程式庫無法滿足您的需求,您可以使用 ML Kit 的自訂TensorFlow Lite建置。例如,您可能想要新增自訂操作。

先決條件

捆綁適用於 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'