讲师中心 微信公众号
AI工具推荐 视频效率加速

C++ 怎么使用 ONNX Runtime C++ API 加载和运行各类机器学习模型

浅磊小哥_2085

浅磊小哥_2085

发布时间:2026-10-30 11:06:47

|

536人浏览过

|

来源于php中文网

原创

必须严格匹配模型输入输出的shape、dtype和name,否则易崩溃或输出全零;Ort::Env需显式构造并设日志级别为WARNING,SessionOptions须预设图优化、线程数及CUDA提供者,输入张量须用std::vector持久化内存并带batch维,输出名与shape须动态获取验证。

c++ 怎么使用 onnx runtime c++ api 加载和运行各类机器学习模型

能直接用,但必须严格匹配模型输入输出的 shape、dtype 和 name,否则大概率崩溃或输出全零——这不是 API 用错了,是张量对不上。

Ort::Env 和 Ort::SessionOptions 怎么初始化才不踩坑

Ort::Env 必须显式构造,不能依赖全局静态对象;Ort::SessionOptions 必须在创建 Ort::Session 前完成所有关键配置。Windows 下链接失败八成是 onnxruntime.lib 没正确导入,或 ABI 不一致(比如项目用 /MD 而库是 /MT 编译的)。

  • Ort::Env env(ORT_LOGGING_LEVEL_WARNING, "my_app") —— 建议设为 ORT_LOGGING_LEVEL_WARNING,避免调试日志刷屏
  • session_options.SetGraphOptimizationLevel(ORT_ENABLE_ALL) —— 图优化必须手动开启,否则默认不启用常量折叠、算子融合等关键优化
  • CUDA 提供者注册必须在 Ort::Session 构造前:调用 OrtSessionOptionsAppendExecutionProvider_CUDA(options, 0)
  • Linux 下记得确保 LD_LIBRARY_PATH 包含 libonnxruntime.so 所在路径

输入张量怎么构造才不会 crash

ONNX Runtime 不接管内存,Ort::Value::CreateTensor 的第四个参数(数据指针)必须在整个 Run() 调用期间有效。栈变量传指针、std::vector 临时对象取 .data() 后立刻析构,都会导致段错误。

C++14
C++14

C++14 对 C++11 的修正与增强版本,适合旧系统维护和较老工具链兼容。

下载
  • 用 std::vector<float></float> 分配输入数据,再传 .data() —— 安全且可控
  • 输入 shape 必须带 batch 维度,哪怕只推一个样本:{1, 3, 224, 224},不是 {3, 224, 224}
  • 图像默认排布是 NCHW;若模型导出为 NHWC,要么预处理时重排,要么用 Ort::MemoryInfo::CreateCpu(...) 显式指定 layout 并自行保证内存布局
  • 归一化参数顺序错(比如 BGR/RGB 颠倒、均值方差套错通道)会导致输出全零,不是报错,极难排查

怎么安全读取输出结果

别猜输出名、别硬编码 shape、别假设 dtype。ONNX 模型的输出节点名和结构由导出时决定,和 Python 端可能不一致。

立即学习“C++免费学习笔记(深入)”;

  • 先用 output_values[0].GetTensorTypeAndShapeInfo().GetShape() 拿真实 shape,比如分类任务常见 {1, 1000},生成任务可能是 {1, 128}
  • 用 GetElementType() 确认类型:ONNX_TENSOR_ELEMENT_DATA_TYPE_FLOAT 对应 .GetTensorData<float>()</float>,ONNX_TENSOR_ELEMENT_DATA_TYPE_INT64 对应 .GetTensorData<int64_t>()</int64_t>
  • 输出名不能写死为 "output",得用 session.GetOutputName(0, allocator) 动态获取,或提前用 netron 查看模型结构
  • 输出张量数据指针直接解引用前,务必确认长度:用 output_values[0].GetTensorTypeAndShapeInfo().GetElementCount() 校验

最麻烦的从来不是加载模型,而是把输入数据从「看起来像」变成「完全匹配模型期望」——shape、dtype、layout、归一化、name,五者缺一不可。少查一项,就可能花半天在空输出或崩溃上兜圈子。

热门AI工具

更多
UP简历
UP简历 Hot

一款AI办公效率工具,主要用于基于AI技术的免费在线简历制作工具,适合需要提升相关任务效率的用户。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

小橡皮AI

一款AI工具,主要用于小橡皮一键去AI味,将机器味重的内容改得更有活人感,防止因为 AI 内容被平台打上 AI 标签,以及限流,适合需要提升相关任务效率的用户。

Atoms
Atoms Hot

Atoms是一款AI智能体工具,第一支自动构建真实业务的 AI 团队。

二狗PPT
二狗PPT Hot

一款AI演示文稿工具,主要用于专为中式职场打造的AI PPT生成工具,适合需要提升相关任务效率的用户。

蛙蛙写作

一款AI论文写作工具,主要用于超级AI智能写作助手,适合需要提升相关任务效率的用户。

Laper
Laper Hot

Laper是专为编剧、导演和制片人推出的 AI 原生剧本创作工具。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

相关专题

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

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

2268

2024.03.14

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

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

1039

2024.03.14

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

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

447

2024.03.14

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

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

327

2024.03.14

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

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

426

2024.03.14

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

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

600

2024.03.14

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

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

1429

2024.03.21

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

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

1237

2024.03.22

Vite 项目如何打包和部署
Vite 项目如何打包和部署

Vite项目打包部署专题汇集生产构建、产物预览、输出目录、资源路径、Nginx与GitHub Pages部署及常见白屏和路由404问题的已审核文章,帮助读者完成上线流程。

40

2026.10.28

热门下载

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

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
C++ Core Guidelines
C++ Core Guidelines

共0课时 | 0人学习

C++ Reference
C++ Reference

共0课时 | 0人学习

C++ 官方标准说明
C++ 官方标准说明

共0课时 | 0人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn