콘솔로 이동

맞춤 텐서플로우 라이트 빌드 사용plat_android

숙련된 ML 개발자이며 기본 제공되는 텐서플로우 라이트 라이브러리가 요구사항에 맞지 않는 경우 ML Kit와 함께 맞춤 텐서플로우 라이트 빌드를 사용할 수 있습니다. 예를 들어 커스텀 작업을 추가할 수 있습니다.

기본 요건

Android용 맞춤 텐서플로우 라이트 번들로 묶기

다음과 같이 텐서플로우 라이트 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 파일이 생성됩니다. 맞춤 텐서플로우 라이트 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에서 텐서플로우 라이트를 맞춤 버전으로 재정의합니다.

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