Utiliser une version personnalisée de TensorFlow Lite

Si vous êtes un développeur ML expérimenté et que la bibliothèque TensorFlow Lite prédéfinie ne répond pas à vos besoins, vous pouvez utiliser une version TensorFlow Lite personnalisée avec ML Kit. Par exemple, vous pouvez ajouter des opérations personnalisées.

Prérequis

Regrouper un fichier TensorFlow Lite personnalisé pour Android

Créez le fichier 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

Un fichier AAR est alors généré dans bazel-genfiles/tensorflow/lite/java/. Publiez l'AAR Tensorflow Lite personnalisé dans votre dépôt Maven local :

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

Enfin, dans votre application build.gradle, remplacez TensorFlow Lite par votre version personnalisée :

implementation 'org.tensorflow:tensorflow-lite:0.1.100'