Sử dụng bản dựng tuỳ chỉnh cho TensorFlow Lite

Nếu là một nhà phát triển máy học giàu kinh nghiệm và thư viện TensorFlow Lite tạo sẵn không đáp ứng nhu cầu của bạn, bạn có thể sử dụng bản dựng TensorFlow Lite tuỳ chỉnh bằng Bộ công cụ máy học. Ví dụ: bạn có thể muốn thêm các thao tác tuỳ chỉnh.

Điều kiện tiên quyết

• Môi trường xây dựng TensorFlow Lite đang hoạt động

Gói TensorFlow Lite tuỳ chỉnh cho Android

Tạo 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

Thao tác này sẽ tạo một tệp AAR trong bazel-genfiles/tensorflow/lite/java/. Phát hành AAR Tensorflow Lite tuỳ chỉnh vào kho lưu trữ Maven cục bộ:

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

Cuối cùng, trong ứng dụng build.gradle, hãy ghi đè TensorFlow Lite bằng phiên bản tuỳ chỉnh:

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