// 1. 创建环境(全局一次)
Ort::Env env{ORT_LOGGING_LEVEL_WARNING, "test"};
// 2. 配置会话选项(可启用 graph optimization / intra-op thread pool)
Ort::SessionOptions session_options;
session_options.SetIntraOpNumThreads(4);
session_options.SetGraphOptimizationLevel(GraphOptimizationLevel::ORT_ENABLE_ALL);
// 3. 创建会话(加载模型)
Ort::Session session{env, L"resnet18.onnx", session_options};
// 4. 获取输入输出信息(用于分配内存和校验 shape)
auto input_node_names = session.GetInputNames();
auto output_node_names = session.GetOutputNames();
auto input_shape = session.GetInputTypeInfo(0).GetTensorTypeAndShapeInfo().GetShape();
// 5. 准备输入张量(例如 float32 的 1x3x224x224 图像)
std::vector input_tensor_values(1 3 224 * 224, 0.5f); // dummy data
std::vector input_tensor_shape{1, 3, 224, 224};
auto memory_info = Ort::MemoryInfo::CreateCpu(OrtArenaAllocator, OrtMemTypeDefault);
Ort::Value input_tensor = Ort::Value::CreateTensor(
memory_info, input_tensor_values.data(), input_tensor_values.size(),
input_tensor_shape.data(), input_tensor_shape.size());
// 6. 推理执行
auto output_tensors = session.Run(Ort::RunOptions{nullptr},
input_node_names.data(), &input_tensor, 1,
output_node_names.data(), 1);
// 7. 读取输出(例如分类 logits)
float* output_data = output_tensors[0].GetTensorMutableData();
int64_t output_count = output_tensors[0].GetTensorTypeAndShapeInfo().GetElementCount();
c++如何实现一个简单的神经网络推理框架_c++ ONNX Runtime集成【AI】
推荐集成ONNX Runtime而非手写推理框架,因其专为高效推理设计,支持多硬件后端、量化、图优化及原生C++ API,只需加载ONNX模型并执行推理即可。
用 C++ 实现一个“从零开始”的完整
神经网络
推理框架工程量大、易出错,不推荐;实际项目中更高效可靠的做法是集成成熟推理引擎——
ONNX Runtime
是目前最主流、跨平台、高性能且对 C++ 友好的选择。
为什么选 ONNX Runtime 而不是手写?
它不是模型训练框架,而是专为**高效推理**设计的运行时:支持 CPU/GPU(CUDA / DirectML / Vulkan)、量化、图优化、多线程、内存复用,并原生提供稳定 C++ API。你只需加载 ONNX 模型(由 PyTorch/TensorFlow 导出),喂入数据,拿到输出,无需管理张量内存、算子实现或调度逻辑。
快速集成步骤(Windows/Linux/
mac
OS 通用)
前提:
已有一个导出好的 .onnx 模型(例如 resnet18.onnx)
下载预编译库:访问
ONNX Runtime 官方 Release 页面
,选对应系统和硬件后端(如 onnxruntime-win-x64-gpu-1.18.0.zip)
解压后,将 include/ 头文件目录和 lib/onnxruntime.lib(Windows)或 lib/libonnxruntime.so(Linux)加入你的 C++ 工程
链接时添加 onnxruntime 库,启用 C++17(必需)
代码结构精简清晰:环境 → 会话 → 输入/输出绑定 → 运行
C++ 核心调用示例(无 OpenCV 依赖,纯推理)
以下是最小可运行片段(省略错误检查,实际需加 try/catch 和 status 判空):
Manus(Windows)
Manus 1.5.6.0是一款出色的Windows人工智能助手,为你解锁PC端的AI新玩法。无论是解答各类疑问、大幅提升工作效率,还是制作趣味视听内容,它都能轻松胜任。更值得一提的是,应用每天赠送300免费积分(精简版同样适用),例如使用GPT-Image-1生成一张图片仅需约60积分。立即下载Manus,畅享AI带来的便捷与乐趣!
下载
立即学习
“
C++免费学习笔记(深入)
”;
常见问题与建议
模型输入预处理必须和训练时完全一致
:归一化(如 ImageNet 的 mean=[0.485,0.456,0.406], std=[0.229,0.224,0.225])、通道顺序(CHW vs HWC)、插值方式等,建议用 OpenCV 或 stb_image + 手动排列
GPU 推理需安装 CUDA/cuDNN 并使用 onnxruntime-gpu 包;CPU 版默认开启 AVX2/SSE 优化,无需额外配置
若需动态 batch 或 shape(如 -1 in shape),确保模型导出时已设 dynamic_axes,且 ONNX Runtime 版本 ≥1.14
性能瓶颈常在数据搬运(host ↔ device)和预处理,而非推理本身;可用 Ort::IoBinding 绑定 GPU 内存减少拷贝
基本上就这些。不需要造轮子,用好 ONNX Runtime 就是当前 C++ AI 推理最务实的选择。
