Pt模型导出Openvino模型

目录

一、安装openvino

1、激活虚拟环境(我的虚拟环境ocr_detect)

2、下载openvino工具

二、导出openvino模型

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

2、使用方法

三、参考博客

---

一、安装openvino

1、激活虚拟环境(我的虚拟环境ocr_detect)

activate ocr_detect

2、下载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模型

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

"D:\installed software\Anaconda\envs\ocr_detect\Lib\site-packages\openvino\tools\mo\mo_onnx.py"

2、使用方法

python mo_onnx.py文件路径 --input_model(被转换的onnx模型路径) --output_dir(输出的模型文件夹路径)

示例

python "D:\installed software\Anaconda\envs\ocr_detect\Lib\site-packages\openvino\tools\mo\mo_onnx.py" --input_model "C:\Users\admin\Desktop\IDmodel20241209\Moedel_ncnn20241209_epochs200\openvino\crnn200_20241205.onnx" --output_dir "C:\Users\admin\Desktop\IDmodel20241209\Moedel_ncnn20241209_epochs200\openvino"

三、参考博客

用OpenVino将onnx模型转换成xml模型_pytorch onnx转xml-优快云博客