RK3562/3588系列之6—yolov5模型的部署
yolov5是使用非常广泛的深度学习模型之一。同时,RK官方资料中也有非常多的资料。
本博客结合这些资料,进行yolov5在RK3588开发板部署的实践介绍。
整体过程包括:
- 模型的训练;
- 训练好的模型(如.pt)导成onnx格式;
- onnx模型转为RKNN格式;
- NPU读取模型进行推理。
代码执行建议通过terminal命令行进行。
1.yolov5模型训练
注意要使用 RK官方github yolov5 。虽然它也是基于https://github.com/ultralytics/yolov5 ,但也有一些变化。
README中有简单介绍:
直接使用