커스텀 TensorFlow Lite 빌드 사용

숙련된 ML 개발자이며 사전 빌드된 TensorFlow Lite 라이브러리가 요구사항에 맞지 않는 경우 ML Kit를 통해 커스텀 TensorFlow Lite 빌드를 사용할 수 있습니다. 예를 들어 커스텀 작업을 추가할 수 있습니다.

기본 요건

• 작동하는 TensorFlow Lite 빌드 환경

Android용 커스텀 TensorFlow Lite 번들로 묶기

다음과 같이 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

그러면 bazel-genfiles/tensorflow/lite/java/에 AAR 파일이 생성됩니다. 커스텀 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'