1.在yolov5上将pt 转 onnx
首先将之前训练好的best.pt放到yolov5目录下
python export.py --weights best.pt --img 640 --batch 1 --include onnx
可以看到生成了best.pt
《yolov5 将训练的best.pt 转成onnx再转成rknn部署到rv1109》
发布时间 2023-04-14 11:16:41作者: 一个不知道干嘛的小萌新
1.在yolov5上将pt 转 onnx
首先将之前训练好的best.pt放到yolov5目录下
python export.py --weights best.pt --img 640 --batch 1 --include onnx
可以看到生成了best.pt