TensorFlowLite:将TensorFlow训练好的模型迁移到Android APP上

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/casgj16/article/details/83118570

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

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

猜你喜欢

转载自blog.csdn.net/casgj16/article/details/83118570