Android Tensorflow Lite C++ .SO 库未在运行时链接

Posted 2023-05-06

【中文标题】Android Tensorflow Lite C++ .SO 库未在运行时链接

【英文标题】：Android Tensorflow Lite C++ .SO libraries not linked on runtime

【发布时间】：2020-08-17 20:48:56

【问题描述】：

我正在尝试在 android 设备上原生使用 C++ 构建 Tensorflow lite。我为每个架构构建了一个 .so 的 tensorflow 文件并放入 jniLibs 文件夹中。这是我的 Cmake 文件：

set(pathToProject /ih/user/project/NativeTfLite/app)

set(libs $pathToProject/src/main/jniLibs)
add_library(libtensorflowLite SHARED IMPORTED
        )
set_target_properties(libtensorflowLite PROPERTIES IMPORTED_LOCATION
        $libs/$ANDROID_ABI/libtensorflowLite.so)

find_library( 
        log-lib
        log )

add_library(
        native-lib
        SHARED
        native-lib.cpp)

target_include_directories(native-lib PRIVATE
        $lib/include)

target_link_libraries( # Specifies the target library.
        native-lib.     #Problem is here when linking native-lib with libtensorflowlite
        libtensorflowLite
        $log-lib )

在编译期间会找到这些文件。但是在 target_link_libraries 行中，两个库 libtensorflowlite 和 native-lib 的链接存在运行时崩溃并出现以下错误：

java.lang.UnsatisfiedLinkError: dlopen failed: library "/Users/ih/project/cameraFrames/NativeTfLite/app/src/main/jniLibs/arm64-v8a/libtensorflowLite.so" 找不到

此外，.SO 文件未链接到 APK。

这是我的 build.gradle 文件：

apply plugin: 'com.android.application'
apply plugin: 'kotlin-android'
apply plugin: 'kotlin-android-extensions'

android 
    compileSdkVersion 29
    buildToolsVersion "29.0.3"

    defaultConfig 
        applicationId "com.proj.nativetflite"
        minSdkVersion 24
        targetSdkVersion 29
        versionCode 1
        versionName "1.0"

        testInstrumentationRunner "androidx.test.runner.AndroidJUnitRunner"
        sourceSets 
            main 
                jniLibs.srcDir 'src/main/jniLibs'
                jniLibs.srcDirs = ["src/main/jniLibs"]
            
        
        externalNativeBuild 
            cmake 
                cppFlags "-std=c++11"
                arguments '-DCMAKE_VERBOSE_MAKEFILE=ON'
            
        
        ndk 
            abiFilters  "armeabi-v7a", "x86" , "arm64-v8a"
        
    

    buildTypes 
        release 
            minifyEnabled false
            proguardFiles getDefaultProguardFile('proguard-android-optimize.txt'), 'proguard-rules.pro'
        
    

    externalNativeBuild 
        cmake 
            path "src/main/cpp/CMakeLists.txt"
            version "3.10.2"
        
    


dependencies 
    implementation fileTree(dir: 'jniLibs', include:  '**/*.so')

    implementation "org.jetbrains.kotlin:kotlin-stdlib-jdk7:$kotlin_version"
    implementation 'androidx.appcompat:appcompat:1.1.0'
    implementation 'androidx.core:core-ktx:1.2.0'
    implementation 'androidx.constraintlayout:constraintlayout:1.1.3'
    testImplementation 'junit:junit:4.12'
    androidTestImplementation 'androidx.test.ext:junit:1.1.1'
    androidTestImplementation 'androidx.test.espresso:espresso-core:3.2.0'

这是什么原因造成的？

【参考方案1】：

在 AGP 4.0 之前，您需要使用 jniLibs: https://developer.android.com/studio/projects/gradle-external-native-builds#jniLibs 显式打包预构建库。请注意，文档说您不需要为 CMake 声明的任何内容执行此操作，但我相当确定文档在这里是错误的（我已联系文档所有者，看看我们是否需要修复它，因为它 对于 4.0 是正确的）。

这在 4.0 中应该是不必要的，但还没有达到稳定。

【参考方案2】：

虽然我认为 Dan Albert 的回答应该被接受，但我想补充一点，我得到了完全相同的错误，因为我没有为我的 tflite 库设置正确的路径。

我将 .so 文件保存在 src/main/cpp/tflite/lib/$ANDROID_ABI/ 中，所以我需要在 build.gradle 中添加：

main 
    jniLibs.srcDirs = ['src/main/cpp/tflite/lib']

另外，关于确保 .so 的 SONAME 正确，可能值得查看 this answer（也来自 Dan）。