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