pt转blob

yolo系列训练出来的模型为.pt格式文件

1.pt转onnx

通过export.py文件将pt文件转为onnx文件
export.py文件使用yolov5项目中提供的文件即可
文件下载地址:

https://github.com/Shaoyc/FeatureRecognition/tree/master/References

转换命令:

python export.py --simplify --opset 12 --include onnx --batch-size 1 --imgsz 640 --weights yolov5.pt

2.onnx转IR

通过mo.py文件将onnx文件转为xml,bin,mapping三个文件
mo.py文件使用openvino项目中提供的文件即可
文件下载地址:

https://github.com/Shaoyc/FeatureRecognition/tree/master/References

转换命令:

python mo.py --input_model ./yolov5.onnx --output_dir ./ --input_shape [1,3,640,640]  --scale 255 

3.IR转blob

通过BlobConverter网站将ir文件转换为blob文件
BlobConverter网址:

http://blobconverter.luxonis.com/

Choose model source选项选择openvino model
需要上传ir模型的xml和bin文件
Shaves选择6
转换时长根据模型大小而定,建议先用小模型测试转换时长。

你可能感兴趣的:(#,python,深度学习,开发语言)