进行yolov5自定义模型训练后,形成.Pt文件,要想在rk3588开发板上部署yolov5的训练模型,需要将.pt文件先转化成onnx模型,最后再转成rknn完成部署
一、.pt文件转onnx文件
在配置好的环境中,在项目路径下输入:
python export.py --weights runs/train/exp18/weights/best.pt --img-size 640 --batch-size 1(注意:一定是在你训练成功的环境,不放心就用prompt)
其中weights后面是你训练得到的best.pt文件的路径
转化完成后即可得到.onnx文件