YOLOv5导出OpenVINO学习笔记

最后的结果就是可以在cpu上跑AI程序了。

1. 首先正常下载yolov5，并安装所需要的环境，在此不再赘述。
2. 安装onnx

pip install onnx

3. 在Ubuntu18上安装OpenVINO，官网在此

wget https://apt.repos.intel.com/intel-gpg-keys/GPG-PUB-KEY-INTEL-SW-PRODUCTS.PUB
sudo apt-key add GPG-PUB-KEY-INTEL-SW-PRODUCTS.PUB
echo "deb https://apt.repos.intel.com/openvino/2022 bionic main" | sudo tee /etc/apt/sources.list.d/intel-openvino-2022.list
sudo apt update
apt-cache search openvino

此时如果出现了一大堆openvino相关的库，那么就可以开始安装了，默认安装路径为/opt/intel/openvino_<VERSION>.<UPDATE>.<PATCH>
直接安装最新版：

sudo apt install openvino

安装指定版本：

sudo apt-cache search openvino

sudo apt install openvino-<VERSION>.<UPDATE>.<PATCH>
如
sudo apt install openvino-2022.1.0

查看版本

apt list --installed | grep openvino

卸载

sudo apt autoremove openvino

后边还有一些步骤，我就不再写了，因为我也不知道有啥用，到这里OpenVINO就安装完成了。
然后似乎还要下边这个命令，但是运行时它自动给我安上了。

pip install openvino

导出

python export.py --weight ./weights/yolov5s.pt --include openvino

碰到了这样的问题

OpenVINO: export failure: libpython3.8.so.1.0: cannot open shared object file: No such file or directory

用下边这个方法解决了

sudo apt install libpython3.8

他似乎没有去conda的环境下搜索库，而是去本机搜索了。另一个解决方案。

但是导出完之后，我这么运行

python detect.py --weight ./weights/yolov5s_openvino_model/yolov5s.xml

发现gpu还是有波动，可是明明输出的时候写的是device=cpu啊。
不理解