Android Studio中TensorFlow Lite头文件找不到问题求助

问题详情 #

在Android Studio中集成TensorFlow Lite库时，Native层代码（如Native-lib.cpp及相关头文件）出现头文件找不到的编译错误。以interpreter.h为例，其中引用的allocation.h实际存在于路径include/tf-lite-api/tensorflow_src/tensorflow/lite/allocation.h，但编译时仍提示文件不存在。项目基于Whisper的Android TFLite示例，尝试修改native库和CMake路径后问题未解决。

相关代码片段 #

interpreter.h中的头文件引用（报错行）

#include <stddef.h>
#include <stdint.h>

#include <atomic>
#include <complex>
#include <cstdio>
#include <cstdlib>
#include <functional>
#include <map>
#include <memory>
#include <string>
#include <utility>
#include <vector>

#include "include/tf-lite-api/tensorflow_src/tensorflow/lite/allocation.h" // 此处报错：文件找不到
#include "include/tf-lite-api/tensorflow_src/tensorflow/lite/c/common.h"  // IWYU pragma: export
#include "include/tf-lite-api/tensorflow_src/tensorflow/lite/core/api/error_reporter.h"
// 其余头文件引用省略

Native-lib.cpp中的头文件引用

#include <jni.h>
#include <string>

#include <android/asset_manager.h>
#include <android/asset_manager_jni.h>
#include <cstdio>
#include <android/log.h>
#include "include/tf-lite-api/tensorflow_src/tensorflow/lite/core/interpreter.h"
#include "include/tf-lite-api/tensorflow_src/tensorflow/lite/kernels/register.h"
#include "include/tf-lite-api/tensorflow_src/tensorflow/lite/model.h"
// 其余头文件引用省略

build.gradle中的CMake配置

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

项目结构说明

• 项目中include/tf-lite-api路径确实存在，包含TensorFlow Lite的源码头文件
• 目标文件allocation.h在指定路径下，但Android Studio未识别该文件路径

解决方案 #

1. 配置CMakeLists.txt的头文件搜索路径 #

在src/main/cpp/CMakeLists.txt中添加头文件搜索目录，让编译器能定位到TensorFlow Lite的头文件根路径：

# 添加包含tf-lite-api的include目录到搜索路径
include_directories(${CMAKE_CURRENT_SOURCE_DIR}/include)

CMAKE_CURRENT_SOURCE_DIR指向CMakeLists.txt所在的src/main/cpp目录，确保编译器能搜索到include/tf-lite-api下的所有头文件。

2. 修正头文件引用路径 #

所有头文件中的引用需要基于CMake配置的搜索路径调整，避免使用冗余的全路径：

• 将interpreter.h中的#include "include/tf-lite-api/tensorflow_src/tensorflow/lite/allocation.h"修改为：

  #include "tf-lite-api/tensorflow_src/tensorflow/lite/allocation.h"
  

• 更规范的方式是调整include目录到tf-lite-api层级：

  include_directories(${CMAKE_CURRENT_SOURCE_DIR}/include/tf-lite-api)
  

  之后头文件引用可简化为官方标准格式：

  #include "tensorflow/lite/allocation.h"
  

3. 确保CMake正确链接TensorFlow Lite库 #

如果是基于源码构建TFLite，需在CMakeLists.txt中添加源码目录并链接对应库：

# 添加TensorFlow Lite源码子目录
add_subdirectory(${CMAKE_CURRENT_SOURCE_DIR}/include/tf-lite-api/tensorflow_src/tensorflow/lite)
# 链接TensorFlow Lite库到你的native库
target_link_libraries(你的Native库名称 PRIVATE tensorflow-lite)

4. 清理缓存并重建项目 #

点击Android Studio顶部菜单：

• Build -> Clean Project
• Build -> Rebuild Project
  清理IDE缓存后重新编译，解决缓存导致的路径识别问题。

5. 验证CMake路径配置 #

当前build.gradle中配置的CMake路径src/main/cpp/CMakeLists.txt是正确的，无需修改CMakeLists.txt的位置，只需确保该文件实际存在且配置正确。

内容的提问来源于stack exchange，提问作者Aniruddh