1.在yolov5上将pt 转 onnx
首先将之前训练好的best.pt放到yolov5目录下
python export.py --weights best.pt --img 640 --batch 1 --include onnx
可以看到生成了best.pt
1.在yolov5上将pt 转 onnx
首先将之前训练好的best.pt放到yolov5目录下
python export.py --weights best.pt --img 640 --batch 1 --include onnx
可以看到生成了best.pt