使用pytorch输出onnx模型并转换成openvino的IR模型
经过好几天的试错,终于成功转换出模型了,出错主要问题是各种包版本不一致的问题。
安装pytorch/torchvision
安装环境和包版本:
Windows 10
torch:1.5.1;
torchvision:0.6.1;
python:3.7.6
onnx:1.7.0
openVino:Anaconda3-2020.02-Windows-x86_64.exe
tensorflow>=1.2.0,<2.0.0
mxnet>=1.0.0,<=1.3.1
networkx=2.0
原创
2020-07-13 17:20:44 ·
3502 阅读 ·
6 评论