本教程是为yolov5s配置openvino,其他网络仅供参考
0. 官方安装教程和要求
官网安装教程链接https://docs.openvino.ai/cn/latest/openvino_docs_install_guides_installing_openvino_windows.html
安装要求:
其中python要求3.6-3.8,可以用anaconda新建一个虚拟环境安装python3.7
1.在官网下载openvino_toolkit
官网下载链接:https://www.intel.com/content/www/us/en/developer/tools/openvino-toolkit/download.html
注意这里要下载2021版本的,如果安装了2020版本,在使用openvino将.onnx
转换为IR
文件时,执行命令python mo.py --input_model=best.onnx --output_dir=Myonnx_IR --model_name=yolowheel --scale=255 --data_type=FP16
可能会出现类似下面的错误
[ ERROR ] operands could not be broadcast together with shapes (4,) (0,)
[ ERROR ]
[ ERROR ] It can happen due to bug in custom shape infer function <function UpsampleOp.upsample_infer at 0x000001BEBC1976A8>.
[ ERROR ] Or because the node inputs have incorrect values/shapes.
[ ERROR ] Or because input shapes are incorrect (embedded to the model or passed via --input_shape).
[ ERROR ] Run Model Optimizer with --log_level=DEBUG for more information.
[ ERROR ] Exception occurred during running replacer "REPLACEMENT_ID" (<class 'extensions.middle.PartialInfer.PartialInfer'>): Stopped shape/value propagation at "Resize_118" node.
For more information please refer to Model Optimizer FAQ (https://docs.openvinotoolkit.org/latest/_docs_MO_DG_prepare_model_Model_Optimizer_FAQ.html), question #38.
根据https://www.51sjk.com/b70b281860/猜测可能是2020版本的openvino还不支持yolov5中的部分算子
2.下载完成后直接打开文件
如果在安装过程中出现闪退问题,是系统不兼容导致,在文件属性中开启兼容模式,开启后再次打开即可
- 一直点击next即可,可以修改安装路径
- 可能出现一些警告,直接next即可
- 安装完成
3.配置环境变量
按下图配置环境变量,注意路径需要换成自己对应的安装路径
如果提示环境变量过长,可以在系统环境变量中按上图的内容输入OpenVINO路径,四个路径用分号分隔,然后在系统变量中的Path中添加 %OpenVINO%,如下图
4.配置Model Optimizer
进入目录D:\Program\intel\openvino_2021\deployment_tools\model_optimizer\install_prerequisites
(根据自己的安装路径对应)
打开cmd,激活anaconda环境(python3.6-3.8),cmd中执行install_prerequisites.bat
命令运行。
等待安装完成。
5.运行demo
进入目录D:\Program\intel\openvino_2021\deployment_tools\demo
在anaconda环境中运行demo_security_barrier_camera.bat
参考:
https://blog.youkuaiyun.com/qq_41251963/article/details/121406569
https://blog.youkuaiyun.com/m0_57055443/article/details/122747401
YOLOv5转openvino并进行部署
https://blog.youkuaiyun.com/gcf_uinque/article/details/105827378
https://blog.youkuaiyun.com/weixin_42586726/article/details/125198259