YOLOv5的模型转换至openvino格式

1、训练YOLOv5模型后转换成onnx格式;

python models/export.py --weights yolov5s.pt --img 640 --batch 1

对于YOLOv5-2.0与3.0,将权重放到models文件夹下;4.0放在外面即可
2、在window端以管理员身份进入命令行:

C:\Windows\System32\cmd.exe
cd C:\Program Files (x86)\Intel\openvino_2021.1.110\deployment_tools\model_optimizer

3、将训练好的权重放到此文件夹下,运行以下指令:

python mo.py --input_model last.onnx --output_dir E:\result --input_shape [1,3,640,640] --data_type FP16

YOLOv5的模型转换至openvino格式_第1张图片

你可能感兴趣的:(YOLOv5的模型转换至openvino格式)