在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')
至此,所有配置已经完成。