Pytorch 转 OpenVINO 的教程

比赛 sophie ⋅ 于 8个月前 ⋅ 最后回复由 你找不到我 6个月前 ⋅ 1836 阅读

Pytorch转OpenVINO需要先将Pytorch转成onnx,然后再onnx转成ir.
其中Pytorch->ONNX,直接看官方教程就好,网上也有很多例子。
ONNX->ir,看openvino的官方教程:
https://docs.openvinotoolkit.org/latest/_docs_MO_DG_prepare_model_convert_model_Convert_Model_From_ONNX.html
另外,推荐大家看Open model zoo上的转换样例,因为在open model zoo里有很多常用模型的转换,如果小伙伴们使用的模型跟这些相近,稍微改动一下就可以。
open model zoo的地址是:https://github.com/opencv/open_model_zoo
下面具体教大家如何利用open model zoo
1)以下地址可以查看具体有哪些模型:
https://github.com/opencv/open_model_zoo/blob/master/models/public/index.md
打开是这样的,可以看到模型的框架
file
以上面的EfficientNet B0为例来说明。
2)通过model downloader的python脚本来下载对就的模型,如何下载模型看这里:
https://docs.openvinotoolkit.org/latest/_tools_downloader_README.html
只用关注其中的downloader.py就好,以下载EfficientNet B0为例,命令为:
python3 downloader.py --name efficientnet-b0 -o ~/download_models/
其中--name指定的是模型的名称,可以通过python3 downloader.py –print_all来查看所有模型的名称,比如EfficientNet B0对应的就是efficientnet-b0.
-o 指定的是模型的下载目录
3)查看EfficientNet B0的说明:
从(1)中的链接点进去,最终能看到两个文件:
file
Markdown文件是对模型的说明,model.yml是对转换参数的说明,比如看这里:
file
这就是mo转换的参数了。直接用这些参数就可以完成转换了。

微信公众号: 极市平台(ID: extrememart )
每天推送最新CV干货

回复数量: 2
您需要登陆以后才能留下评论!