カスタム TensorFlow Lite ビルドの使用

事前に構築された TensorFlow Lite ライブラリがニーズを満たしていない場合、ML デベロッパーとしての経験が豊富であれば、ML Kit とともにカスタム TensorFlow Lite ビルドを使用できます。たとえば、カスタム オペレーターを追加することができます。

前提条件

カスタム TensorFlow Lite for Android のバンドル

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'