ใช้บิลด์ TensorFlow Lite แบบกำหนดเอง

หากคุณเป็นนักพัฒนา ML ที่มีประสบการณ์และไลบรารี TensorFlow Lite ที่สร้างไว้ล่วงหน้าไม่ตรงกับความต้องการของคุณ คุณสามารถใช้ TensorFlow Lite build แบบกำหนดเองกับ 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/ เผยแพร่ 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'