【目标检测】yolov5模型转换从pytorch到onnx到openvino（部署方式）

1.环境

yolov5环境中有onnx就可以，可以在另外的环境中单独安装openvino。

# yolov5的环境 + onnx
onnx==1.9.0

# openvino的环境
openvino_2021.1.110

2.步骤

# 1. yolov5s pytorch model to onnx model （develop分支下）
python3 models/export.py --weights ./weights/yolov5s.pt --img-size 640 --batch-size 1

# 2. yolov5s onnx to openvino model
python3 /opt/intel/openvino_2021.1.110/deployment_tools/model_optimizer/mo.py --input_model yolov5s.onnx --model_name yolov5s -s 255 --reverse_input_channels --output Conv_551,Conv_503,Conv_455

# 3.1.1 test openvino model with CPU
python3 yolo__openvino_demo.py -m yolov5s.xml -i voc/img/2007_000027.jpg -at yolov5 -d CPU

# 3.1.2 test openvino model with GPU
python3 yolo__openvino_demo.py -m yolov5s.xml -i voc/img/2007_000027.jpg -at yolov5 -d GPU

# 3.2.1 test video openvino model with CPU
python3 yolo__openvino_demo.py -m yolov5s.xml -i 2007_000027.mp4 -a