yolo v5 转rknn 请参考: 下一篇:yolo v7 转rknn
一、训练
1.切换版本:yolo v5-release6.0
git clone https://github.com/ultralytics/yolov5
cd yolov5
git checkout 956be8e642b5c10af4a1533e09084ca32ff4f21f
或者安装Gitkraken6.5.0

2.训练
参考yolov5 初识(ubuntu版)、yolov5 初识(win版)
二、pt2onnx
python export.py --weight="runs/train/exp2/weights/best.pt"
三、onnx2rknn
yolo v5 转rknn 请参考: 下一篇:yolo v7 转rknn
本文介绍了如何从yolov5版本6.0开始,通过切换代码库、训练模型、将PyTorch模型转换为ONNX,然后进一步转为RKNN格式的过程。适合想要将YOLOv5模型移植到嵌入式设备的开发者。
987

被折叠的 条评论
为什么被折叠?



