用OpenVino将onnx模型转换成xml模型

一、使用python或者anaconda创建一个虚拟环境

演示用anacnda创建环境

conda create -n openvino python=3.10

激活环境

activate openvino

二、下载openvino工具包

官方安装教程：Install OpenVINO™ Development Tools — OpenVINO™ documentation

使用清华镜像源下载：

pip install openvino-dev[pytorch] -i https://pypi.tuna.tsinghua.edu.cn/simple/

其中[pytorch]，pytorch是参数，指定输入模型的源深度学习框架，并且是以下一个或多个以“，”分隔的值:caffe, kaldi, mxnet, onnx, pytorch, tensorflow, tensorflow2。 

三、执行转换

找到虚拟环境下的openvino文件夹中的mo_onnx.py文件

方法：

python 执行脚本路径 --input_model 被转换的onnx模型路径 --output_dir 输出的xml模型路径

示例：

python D:\Anaconda\envs\pytorch\Lib\site-packages\openvino\tools\mo\mo_onnx.py --input_model C:\Users\CGW\Desktop\yolov8train\best.onnx --output_dir C:\Users\CGW\Desktop\yolov8train

 

四、结果展示（我是在pytorch虚拟环境安装的openvino）

        感谢支持，如果有更好的方法或者改进请不吝提出！