Если вы опытный разработчик машинного обучения и готовая библиотека TensorFlow Lite вам не подходит, вы можете использовать пользовательскую сборку TensorFlow Lite с помощью ML Kit. Например, вы можете добавить пользовательские операции.
Предпосылки
- Рабочая среда сборки TensorFlow Lite
Сборка пользовательского TensorFlow Lite для 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
Это создаст 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'