C++如何使用C++调用TensorFlow Lite进行模型推理

星降

星降

2026-08-03

491人浏览

原创

需同时链接libtensorflowlite.a及absl、flatbuffers等依赖静态库,并添加-ldl -lpthread(linux)或-lsystem(macos)。

c++如何使用c++调用tensorflow lite进行模型推理

如何在C++项目中正确链接TensorFlow Lite库

直接用 g++ 编译时出现 undefined reference to `tflite::InterpreterBuilder::InterpreterBuilder(...) 这类错误,基本是因为没链接对静态库或符号未导出。TensorFlow Lite官方不提供预编译的动态库,C++ API依赖的是 libtensorflowlite.a 及其依赖的 abslflatbuffers 等子模块静态库。

实操建议:

  • 用 CMake 构建(推荐),通过 add_subdirectory(tflite_dir) 引入源码树,避免手动找库路径
  • 若必须用预编译库,确保同时链接:libtensorflowlite.alibabsl_base.alibabsl_strings.alibflatbuffers.a —— 少一个就会报 undefined symbol
  • Linux 下需额外加 -ldl -lpthread;macOS 需加 -ldl(但系统无 libdl.dylib,得用 -lSystem 替代)
  • 启用 -DTFLITE_ENABLE_XNNPACK=OFF 可跳过 XNNPACK 依赖,适合快速验证基础推理流程

加载.tflite模型文件并检查输入/输出张量

tflite::FlatBufferModel::BuildFromFile() 返回 nullptr 却不报错?常见原因是模型文件路径错误、权限不足,或模型本身被加密/损坏(比如误用 .h5 转出但没调用 converter.experimental_enable_resource_variables = True)。

实操建议:

  • std::ifstream(model_path, std::ios::binary).good() 先确认文件可读
  • 构建后立刻调用 model->GetModel()->version(),正常应返回 3;若为 0,说明 flatbuffer 解析失败
  • 输入张量名不一定叫 input:用 interpreter->tensor(interpreter->inputs()[0])->name 打印真实名称,常为 normalized_input_image_tensor 或带数字后缀
  • 注意维度顺序:TFLite 默认是 NHWC,C++ 中填数据要按 [batch, height, width, channel] 排布,别和 PyTorch 的 NCHW 混了

如何安全地将图像数据送入模型输入张量

调用 interpreter->typed_input_tensor<float>(0)</float> 后 memcpy 崩溃?大概率是内存未对齐或尺寸不匹配。TFLite 对 float32 输入要求 4 字节对齐,且 tensor size 必须等于 dims[0] * dims[1] * dims[2] * dims[3]

C函数速查手册(CHM版)
C函数速查手册(CHM版)

C函数速查手册(CHM版)

下载

实操建议:

  • 不要直接 reinterpret_cast 图像指针——先用 interpreter->input_tensor(0)->bytes 获取所需字节数,再 malloc 对齐内存(用 aligned_alloc(16, size)
  • 图像预处理必须和训练时一致:比如 TFLite MobileNetV2 要求 (pixel - 127.5) / 127.5 归一化,不是 /255.0
  • RGB/BGR 顺序要核对:OpenCV 默认 BGR,而多数 TFLite 模型期望 RGB,漏转会导致识别结果全错
  • 如果输入是 uint8 类型(常见于量化模型),用 interpreter->typed_input_tensor<uint8_t>(0)</uint8_t>,别强转 float 指针

运行推理后如何提取输出结果

调用 interpreter->Invoke() 成功,但输出全是 0 或 nan?除了输入问题,更可能是没等异步执行完成(仅限启用 TFLITE_ENABLE_NNAPI 或 GPU 代理时),或输出 tensor 被模型优化掉了(如只保留 final layer,中间 head 被剪枝)。

实操建议:

  • 务必检查 interpreter->outputs().size(),有些模型有多个输出 tensor(如 detection model 输出 locationcategoryscore),索引写死 [0] 会取错
  • 量化模型输出仍是 int8/uint8,需用 output_tensor->params.scalezero_point 手动反量化:float_val = (int8_val - zero_point) * scale
  • 分类模型常用 softmax 后处理,但 TFLite 模型里往往已固化该算子——此时输出已是概率,勿重复 softmax
  • 调试时用 memcpy 把输出拷到 vector 再用 std::max_element 查最大值索引,比手写循环更可靠

最易被忽略的是模型输入输出的数据类型与内存布局——同一份 .tflite 文件,在不同平台(x86/arm64)、不同构建选项(WITH_RUY / WITHOUT_RUY)下,interpreter->AllocateTensors() 分配的内存行为可能不同,务必在目标设备上实测。

C++免费学习笔记(深入):立即使用
在学习笔记中,你将探索 C++ 的入门与实战技巧!

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

c++

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
c++和c语言的区别有哪些
c++和c语言的区别有哪些

c++和c语言的区别:1、面向对象编程(OOP)支持不同;2、新增特性不同;3、标准库不同;4、编译方式不同;5、命名空间不同等等。想了解更多c++的相关内容,可以阅读本专题下面的文章。

2024.03.14

857

9

c++和python学习顺序推荐
c++和python学习顺序推荐

一般建议先学习C++,再学习Python,因为这样可以逐步从较为底层的编程语言向更高级的语言过渡。想了解更多python的相关内容,可以阅读本专题下面的文章。

2024.03.14

754

6

python和c++学习性价比分析
python和c++学习性价比分析

Python易于学习,广泛应用于Web开发、数据科学和人工智能等领域,但性能较低。C语言性能高,适用于对性能要求较高的场景,如游戏开发和系统编程,但学习曲线陡峭,错误处理复杂。想了解更多python的相关内容,可以阅读本专题下面的文章。

2024.03.14

245

5

c语言和c++一样吗
c语言和c++一样吗

c语言和c++是两种不同的编程语言,虽然有相似之处,但存在显著差异。c语言专注于过程式编程和系统级开发,以简洁、高效著称。c++作为c语言的超集,引入了面向对象编程,增强了代码组织和管理能力,但学习曲线也更陡峭。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2024.03.14

243

5

c语言和c++先学哪个好
c语言和c++先学哪个好

初学者选择学习c语言还是c++语言,需要根据个人学习目标、背景以及编程兴趣和预期应用方向来决定。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2024.03.14

224

5

c语言和c++的区别和联系
c语言和c++的区别和联系

c语言和c++是计算机科学领域应用广泛的编程语言。虽然它们有着相似的基础,但它们在语言类型、语法功能和内存管理方面存在着显著差异。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2024.03.14

470

5

c++软件中文更改教程
c++软件中文更改教程

对于 ide,可通过打开设置,找到语言设置,选择中文,并保存更改。对于非 ide 应用程序,可查找设置或选项,选择语言设置,更改为中文,并保存更改。想了解更多c++的相关内容,可以阅读本专题下面的文章。

2024.03.21

1136

9

python和java和c++学习性价比分析
python和java和c++学习性价比分析

Python以其易学性、丰富的库和活跃的社区而著称,适合数据科学、人工智能和Web开发。Java以其跨平台性、企业级应用开发和Android应用开发而闻名。C++以其底层控制能力、高效性能和游戏开发而著称。选择哪种语言取决于个人兴趣、职业方向和特定需求。想了解更多python和java和c++的相关内容,可以阅读本专题下面的文章。

2024.03.22

884

7

c++和c语言学习顺序推荐
c++和c语言学习顺序推荐

对于初学者,建议先学习C语言,掌握编程基础后再转入C++,便于理解面向对象编程概念。有编程经验者可直接学习C++,快速接触高级编程技术。想了解更多c++和c语言的相关内容,可以阅读本专题下面的文章。

2024.03.25

1062

9

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Qt Creator 项目创建文档
Qt Creator 项目创建文档

共0课时 | 0人学习

RabbitMQ 教程手册
RabbitMQ 教程手册

共0课时 | 0人学习

Linux man-pages 项目
Linux man-pages 项目

共0课时 | 0人学习