rk3588部署yolov5
时间: 2023-10-02 16:04:08 浏览: 282
基于RK3588上部署yolov5s模型源码(实时摄像头检测)+部署说明文档.zip
5星 · 资源好评率100%
要在RK3588上部署yolov5模型,可以按照以下步骤进行操作:
1. 首先,在yolov5文件夹中执行export.py文件,命令如下:
```
python export.py --weights best.pt --img 640 --batch 1 --include onnx --rknpu RK3588
```
2. 然后,下载并解压基于RK3588上部署yolov5s模型的源码文件。这个项目是一个个人毕设项目,经过调试和测试,确保可以运行。你可以在这个项目中找到部署说明文档,以便更详细地了解如何在RK3588上部署yolov5s模型。
3. 在RK3588上下载RKNN_toolkit_2,并打开路径/rknn_model_zoo/models/CV/object_detection/yolo/RKNN_C_demo/RKNN_toolkit_2/rknn_yolo_demo/include/yolo.h,将标签数量改成你自己的值。
通过以上步骤,你就可以在RK3588上成功部署yolov5s模型了。
阅读全文