如果您是經驗豐富的 ML 開發人員,且預先建置的 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'