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

Nếu là một nhà phát triển học máy có kinh nghiệm và thư viện TensorFlow Lite được tạo sẵn không đáp ứng được nhu cầu của bạn, thì bạn có thể sử dụng bản dựng TensorFlow Lite tuỳ chỉnh với Bộ công cụ học máy. 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

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/. Xuất bản 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 của bạn:

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