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