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

最新推荐文章于 2024-04-27 12:07:25 发布

liuz_notes

最新推荐文章于 2024-04-27 12:07:25 发布

阅读量902

点赞数 1

分类专栏：模型部署文章标签：深度学习 python 模型部署 ONNXRuntime DBFace

本文链接：https://blog.csdn.net/liuz_notes/article/details/128492454

版权

模型部署专栏收录该内容

3 篇文章 0 订阅

订阅专栏

系列文章目录

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

上一篇博客中简单介绍了ONNXRuntime推理引擎和DBFace检测模型，这篇博客中将展示如何使用其Python API进行初步验证及推理部署。

1 ONNX模型表示

ONNX（Open Neural Network Exchage）是一个开源机器学习模型数据的存储标准，支持不同的人工智能框架，从而将其训练好的模型转为ONNX中间表示，便于ONNXRuntime进行推理部署。本系列博客均使用Pytorch所训练的模型。

在这里插入图片描述

2 模型转换

Pytorch支持ONNX模型转换，仅需调用torch.onnx.export()即可完成模型转换，并且支持模型中的绝大部分算子，从而能够避免很多自己手写算子的麻烦。具体的算子版本支持情况可以查阅官方文档。
在这里插入图片描述

在这里插入图片描述

模型转换完成后，可以使用NETRON工具可视化ONNX模型的输入输出以及算子节点信息，观察模型算子及运算分支是否转换正确。NETRON的可视化节点可以点击查看算子详情，如下图所示。

在这里插入图片描述

3 Python环境下的推理

经过NETRON可视化验证后的模型不一定能得到预想的正确结果，所以我们需要在Python环境下进行推理，确保ONNX模型能得到与Pytorch模型一致的结果。这里需要用到ONNXRuntime的Python API读取并运行模型。
在这里插入图片描述

onnxruntime.InferenceSession用于获取一个 ONNX Runtime 推理器，其参数是用于推理的 ONNX 模型文件。InferenceSession的 run 方法用于模型推理，其第一个参数为输出张量名的列表，第二个参数为输入值的字典。