转换 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) 指南中的*通用转换参数*部分。