第7篇 c#推理自己训练的yolov5 onnx模型文件
在第2篇完成图像训练后,
yolov5-master\runs\train\exp50\weights目录下会生成best.pt的文件
1.将PT模型文件转onnx:
1、用到export.py
安装依赖包
pip3 install onnx
pip3 install coremltools==4.0
文件路径
进行修改:
2.export.py中的配置
coco128.yaml修改为自己训练的数据集配置文件xx.yaml,参考第2篇的训练方法