转换 ONNX 模型

ONNX 介绍

ONNX 是一种深度学习模型的表示格式,支持 AI 开发人员轻松地在不同框架之间传输模型。就像 PyTorch、Caffe2、Apache MXNet、Microsoft Cognitive Toolkit 和其他许多工具一样,它也是一款非常受欢迎的深度学习工具。

转换 ONNX 模型

本页介绍了如何利用模型优化器将模型从 ONNX 格式转换为 OpenVINO IR 格式。要使用模型优化器,请遵循 安装说明 来安装 OpenVINO 开发工具。

模型优化器流程假设您可以拥有一个可以直接从公共数据模型库下载、或者通过任何支持导出为 ONNX 格式的框架转换而成的 ONNX 模型。

要转换 ONNX 模型,请使用到输入模型 .onnx 文件的路径运行模型优化器:

mo --input_model <INPUT_MODEL>.onnx

不存在 ONNX 特定参数,因此只有无关框架的参数可用于转换您的模型。有关详细信息,请参阅 将模型转换为中间表示 (IR) 指南中的*通用转换参数*部分。

支持的 ONNX 层

有关支持的标准层列表,请参阅 支持的框架层 页面。

其他资源

请参阅 模型转换教程 页面获取一系列教程,了解转换特定 ONNX 模型相关的分步指导。部分示例如下: