文章作者:激萌の小宅
促销:¥0
价格:¥0
配送方式:
购买后立即生效(如购买异常,请联系站长)
付款之后一定要等待自动跳转结束,否则购买可能会失败
-
0 天
有效期
-
0
总销量
-
0
累计评价
YOLOv5 pt转onnx文件
参考文章:《YOLOv5 C#中进行模型预测(.net版)》
进行模型转换时,需要安装一些库,安装完之后,才能执行换行命令,不过yolov5-master可以不必手动安装,程序在进行转换时,如果未检测到onnx,会自动安装,但是yolov5 6.1需要手动安装才行,安装命令如下:
pip install -r requirements.txt coremltools onnx onnx-simplifier onnxruntime openvino-dev tensorflow-cpu
1、生成静态模型
python export.py --weights runs/train/exp/weights/best.pt --include onnx
2、生成动态模型
python export.py --weights runs/train/exp/weights/best.pt --include onnx --dynamic
从生成效果看,动态模型会比静态模型小一点