Yolov8在RK3588上进行自定义目标检测(四)

Xiao_Ya__2024-08-05 8:17

参考

YOLOV8火灾检测模型的边缘端推理

验证rknn模型

1.将转换好的rknn模型上传到板子上，再在板子上安装rknn-toolkit-lite2,将上面的requirements_cp.38-2.0.0b0.txt和同一个项目下的rknn_toolkit_lite2-2.0.0b0-cp38-cp38-linux_aarch64.whl下载下来上传到板子的某个目录下。

复制代码

# 安装 RKNN-Toolkit2
pip install rknn_toolkit_lite2-2.0.0b0-cp38-cp38-linux_aarch64.whl

验证是否安装成功

复制代码

# 进入 Python 交互模式 
python 
# 导入 RKNNLite 类  
from rknnlite.api import RKNNLite

2.在rk3588上修改/rknn_model_zoo/examples/yolov8/model/coco_80_labels_list.txt为自己的标签种类；修改同目录下的dataset.txt为自己数据集中的一张图片；

在/userdata/OCR/rknn_model_zoo/目录下执行以下代码。

复制代码

./build-linux.sh -t rk3588 -a aarch64 -d yolov8

3.执行后会在install/rk3588_linux_aarch64/rknn_yolov8_demo/下生成验证rknn模型程序:rknn_yolov8_demo。执行该程序验证rknn模型：

复制代码

./rknn_yolov8_demo model/yolov8.rknn model/1.jpg

4.运行完成后，会在rknn_yolov8_demo下生成out.png文件。

推理结果

参考链接：

YOLOv8部署在RK3588上_rk3588 yolov8部署-CSDN博客

yolov8目标检测部署瑞芯微rk3588记录_yolov8 rknn-CSDN博客

https://download.csdn.net/blog/column/12698056/139685411