1. 程式人生 > >TensorFlowLite:將TensorFlow訓練好的模型遷移到Android APP上

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')

至此,所有配置已經完成。