openvino-YOLOv5轻量化部署(windows端)

利用openvino框架的YOLOv5模型的轻量化部署,在windows端(linux端同理)

github地址:
链接: https://github.com/wang-yichen/openvino-YOLOv5/tree/main

1.openvino开发工具下载

可查阅官方文档
链接: https://docs.openvino.ai/2024/get-started/install-openvino/install-openvino-windows.html
查看openvino环境的安装部分,本文使用的是c++版本开发YOLOv5部署,使用的是存档安装的方法,具体前往官网下载

链接: https://www.intel.com/content/www/us/en/developer/tools/openvino-toolkit/download.html?PACKAGE=OPENVINO_BASE&VERSION=v_2024_4_0&OP_SYSTEM=WINDOWS&DISTRIBUTION=PIP

下载最新的安装包,如下图所示:

在这里插入图片描述

点击Download下载zip包。

下载好后解压,放到任意位置,向环境变量path中添加如下变量,按照自己的位置修改

D:\Program Files\openvino\openvino_2024\runtime\bin\intel64\Debug
D:\Program Files\openvino\openvino_2024\runtime\bin\intel64\Release
D:\Program Files\openvino\openvino_2024\runtime\3rdparty\tbb\bin

2.python环境安装

这里选择使用conda的虚拟环境安装openvino的python环境

pip install openvino-genai==2024.3.0

根据YOLOv5需要,安装下列安装包

python -m pip install --upgrade pip
pip install onnx
pip install torch
pip install mxnet

先从YOLOv5的环境中将.pt转换为.onnx模型,版本选择10

随后编写如下脚本

from openvino.runtime import Core
from openvino.runtime import serialize
 
ie = Core()
onnx_model_path = r"自己的模型路径.onnx"
model_onnx = ie.read_model(model=onnx_model_path)
# compiled_model_onnx = ie.compile_model(model=model_onnx, device_name="CPU")
serialize(model=model_onnx, xml_path="生成xml的模型路径.xml", bin_path="生成bin模型的路径.bin", version="UNSPECIFIED")

运行后生成.xml和.bin模型,如果报错说明环境安装有问题

3.c++部署

创建新项目,release版本

包含目录添加如下内容

D:\Program Files\openvino\openvino_2024\runtime\bin\intel64\Release
D:\Program Files\openvino\openvino_2024\runtime\include
D:\Program Files\openvino\openvino_2024\runtime\include\openvino
C:\E\Users\wangyichen\opencv\build\include\opencv2
C:\E\Users\wangyichen\opencv\build\include

库目录添加如下内容

C:\E\Users\wangyichen\opencv\build\x64\vc16\lib
D:\Program Files\openvino\openvino_2024\runtime\lib\intel64\Release

链接器附加依赖项添加如下内容(依照自己的opencv版本添加)

opencv_world470.lib
openvino.lib

如下为代码部分,直接添加进自己的项目中

openvino.cpp
openvino.h
main.cpp

如此运行可以得到正确的结果,如果有报错请在评论区提出来

Logo

加入社区!打开量化的大门,首批课程上线啦!

更多推荐