使用自訂 TensorFlow Lite 版本

如果您是經驗豐富的機器學習開發人員，且預先建構的 TensorFlow Lite 程式庫無法滿足您的需求，您可以使用 ML Kit 中的自訂 TensorFlow Lite 版本。舉例來說，您可能會想新增自訂運算。

事前準備

• 可用的 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'