Используйте пользовательскую сборку TensorFlow Lite

Если вы опытный разработчик машинного обучения и встроенная библиотека TensorFlow Lite не соответствует вашим потребностям, вы можете использовать специальную сборку TensorFlow Lite с ML Kit. Например, вы можете добавить собственные операции.

Предварительные условия

Объединение пользовательского TensorFlow Lite для Android

Создайте AAR Tensorflow Lite:

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

Это создаст файл AAR в bazel-genfiles/tensorflow/lite/java/ . Опубликуйте собственный AAR Tensorflow Lite в локальном репозитории 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'