yolo8 RK3588 rknn模型训练和获取

参考：https://blog.csdn.net/Striver_Nwj/article/details/143215393
https://blog.csdn.net/A_l_b_ert/article/details/141610417

数据集准备和转化

voc数据集
格式如下：

转换成yolo所需的数据集：

voc转yolo数据集代码：

训练

💡

1、注意需使用https://github.com/airockchip/ultralytics_yolov8这里的代码
2、把上面链接的源码中ultralytics/nn/modules/conv.py的nn.SiLU()都替换成nn.ReLU()

train.py:

pt模型转RKNN

📌

以下转换都需要在Linux中进行：且安装：
rknn-toolkit2：https://github.com/airockchip/rknn-toolkit2
rknn_model_zoo：https://github.com/airockchip/rknn_model_zoo

pt转onnx

pt2rknn.py

onnx转rknn

把onnx复制到rknn_model_zoo项目中，修改下图内容：

修改完以后执行：

结果在../model/hit.rknn

python预测运行代码

https://git.bwbot.org/publish/rknn3588-yolov8

修改于 2025-07-08 09:27:17

jetson orin nano安装环境

设置850摄像头摄像头