TensorFlowLite:將TensorFlow訓練好的模型遷移到Android APP上
在Android Studio中配置TensorFlowLite
把訓練好的 TensorFlow的 pb模型 移植到Android Studio上,TensorFlow Lite官方的例子中已經給我們展示了,我們其實只需要兩個檔案: libandroid_tensorflow_inference_java.jar、libtensorflow_inference.so。
這兩個檔案我已經放到github上了,大家可以自行下載使用,下載地址:
libandroid_tensorflow_inference_java.jar、libtensorflow_inference.so。
然後首先把訓練好的pb模型放到Android專案中app/src/main/assets下,若不存在assets目錄,則自己新建一個。如圖所示
其次,把剛剛下載的 libandroid_tensorflow_inference_java.jar 檔案放到 app/libs 目下,把libtensorflow_inference.so 放到 app/libs/armeabi-v7a 目錄下,如下圖所示:
然後在app/build.gradle裡進行如下配置:
(i) 在defaultConfig裡新增
multiDexEnabled true
ndk{
abiFilters 'armeabi-v8a','armeabi','armeabi-v7a',"x86"
}
jackOptions {
enabled true
}
(ii) 在android裡新增
sourceSets {
main {
jni.srcDirs = []
jniLibs.srcDirs = ['libs']
}
}
compileOptions {
sourceCompatibility JavaVersion.VERSION_1_8
targetCompatibility JavaVersion.VERSION_1_8
}
(iii) 在dependencies中新增libandroid_tensorflow_inference_java.jar,即:
compile 'de.hdodenhof:circleimageview:2.1.0'
compile files('libs/libandroid_tensorflow_inference_java.jar')
至此,所有配置已經完成。