【模型部署】人脸检测模型DBFace C++ ONNXRuntime推理部署（2）系列文章目录1 C++推理流程2 关键API介绍3 DBFace推理C++代码展示4 推理结果及比较参考资料

65 阅读 0 评论 43 点赞

我是靠谱客的博主还单身丝袜，最近开发中收集的这篇文章主要介绍【模型部署】人脸检测模型DBFace C++ ONNXRuntime推理部署（2）系列文章目录1 C++推理流程2 关键API介绍3 DBFace推理C++代码展示4 推理结果及比较参考资料，觉得挺不错的，现在分享给大家，希望可以做个参考。

概述

系列文章目录

【模型部署】人脸检测模型DBFace C++ ONNXRuntime推理部署（0）
【模型部署】人脸检测模型DBFace C++ ONNXRuntime推理部署（1）
【模型部署】人脸检测模型DBFace C++ ONNXRuntime推理部署（2）

文章目录

系列文章目录
1 C++推理流程
2 关键API介绍
3 DBFace推理C++代码展示
4 推理结果及比较
参考资料

在实际应用中，由于语言的特性，通常选择C++环境进行部署，从而更快地得到推理结果。上一篇博客中已经介绍了ONNX模型转换以及Python环境下的ONNXRuntime推理，而本篇博客将分享C++环境下的ONNXRuntime推理部署流程和关键API，并以DBFace模型为例进行展示。

1 C++推理流程

C++环境下的推理并不像Python那样简单易操作，它需要先定义好相关的环境变量，并通过相应的API从ONNX模型得到节点信息，从而分配内存并创建输入输出Tensor，最后进行推理运算，基本流程及所用的API如下图所示。
在这里插入图片描述

2 关键API介绍

首先创建环境变量并设置好Session选项，所有的API都封装在Ort这个类中。Ort::Env只是用来设置日志级别以及日志输出文件，而关键的设置在Ort::SessionOptions & Ort::Session。
在这里插入图片描述
设置运行的线程数和计算图优化级别，以及模型的路径。

Ort::Env ort_env(ORT_LOGGING_LEVEL_WARNING, “default");
Ort::SessionOptions session_options;
session_options.SetIntraOpNumThreads(1);
session_options.SetGraphOptimizationLevel(GraphOptimizationLevel::ORT_ENABLE_ALL);
Ort::Session session(ort_env, pathWch, session_options);

然后通过以下API得到输入输出节点信息。
在这里插入图片描述

Ort::AllocatorWithDefaultOptions allocator;
const char* input_name = session.GetInputName(0, allocator);
auto input_info = session.GetInputTypeInfo(0).GetTensorTypeAndShapeInfo();
auto input_node_dims = input_info.GetShape();

根据得到的输入输出节点信息，创建Tensor并分配内存。
在这里插入图片描述

auto memory_info = Ort::MemoryInfo::CreateCpu(OrtAllocatorType::OrtArenaAllocator, OrtMemType::OrtMemTypeDefault);
Ort::Value input_tensor;
input_tensor = Ort::Value::CreateTensor<float>(memory_info, in_x.data,target_size,input_dims.data(),input_dims.size())

注意其中in_x.data的类型为std::vector<float>。

最后用此前创建的session和Tensor变量进行推理，得到结果。
在这里插入图片描述

auto output_tensors = session.Run(Ort::RunOptions{ nullptr }, input_node_names.data(), input_tensors.data(), input_node_names.size(), output_node_names.data(), output_node_names.size());
float* out_i = output_tensors[i].GetTensorMutableData<float>();