3270|2

290

帖子

0

TA的资源

一粒金砂(高级)

楼主
 

【Luckfox幸狐 RV1103 Linux 开发板测评】04 使用yolov5转换RKNN推理模型图像识别 [复制链接]

 

RV1103的算力有0.5TOPs,使用它来跑一下模型试试看看。那么就先试跑一下RKNN模型;

一、RKNN模型

RKNN 是 Rockchip npu 平台使用的模型类型,以.rknn后缀结尾的模型文件。瑞芯微提供了将自己模型转换为RKNN模型的工具。

二、准备

1、下载瑞芯微的rknpu2

使用如下指令下载:

git clone https://gitee.com/yanyitech/rknpu2.git

下载成功

 

2、编译

1)设置环境变量

export RK_RV1106_TOOLCHAIN=<SDK目录>/tools/linux/toolchain/arm-rockchip830-linux-uclibcgnueabihf/bin/arm-rockchip830-linux-uclibcgnueabihf

2)设置权限

chmod +x build-linux_RV1106.sh

3)编译

在rknpu2/examples/RV1106_RV1103/rknn_yolov5_demo目录下运行build-linux_RV1106.sh脚本;如图

 

如下成功编译

 

三、测试

1、拷贝编译生成文件

编译完成后,install 目录下会生成一个 rknn_yolov5_demo_Linux 目录,需要将整个目录传输至开发板,我使用ssh传输到开发板;但是直接使用scp文件到开发板,操作不了。因此只能使用先从虚拟机复制到WINDOWS,再从windows复制到开发板;如图

 

 

2、运行模型

直接在根目录运行会提示内存不足,运行不了。因此需要先挂载SD卡。

1)挂载SD卡,先格式化为FAT32格式;直接插在开发板即可(开发板默认已经支持挂载SD卡)

使用df -f查看是否挂载成功;如图,成功挂载

 

2)运行模型

    • 使用指令进入到目录

cd /root/rknn_yolov5_demo_Linux/

    • 使用以下指令给rknn_yolov5_demo加权限

chmod rknn_yolov5_demo

    • 配置环境变量

export LD_LIBRARY_PATH=/root/rknn_yolov5_demo_Linux/lib

    • 执行模型

./rknn_yolov5_demo ./model/RV1106/yolov5s-640-640.rknn model/bus.jpg

运行成功,如图

 

3、查看结果

1)先下载官方标记的工具:mark.zip_免费高速下载|百度网盘-分享无限制 (baidu.com)

2)下载之后,将以下数据复制到上述下载得工具目录中data.txt中。同时将rknn_yolov5_demo/model/路径下的bus.jpg图片复制到mark目录

person @ (212 242 285 509) 0.819550 person @ (110 237 217 523) 0.817728 person @ (478 232 559 523) 0.786894 bus @ (101 141 559 445) 0.693319 person @ (78 337 122 518) 0.406052

3)标记

python3 mark.py bus.jpg

成功标记:

 

4)最终原图和标记图。

原图:

 

推理的图:

 

本来还想测试一下,水果和动物的图,奈何128Mflash 太小,下次再测试一下。

最新回复

隔壁我拿的是RV1106,就很舒服了,256的nanoflash   详情 回复 发表于 2024-1-24 16:44
点赞 关注
个人签名

阅读改变人生

 
 

回复
举报

365

帖子

0

TA的资源

版主

沙发
 

隔壁我拿的是RV1106,就很舒服了,256的nanoflash

点评

羡慕啊,我试试SD卡启动应该会好一点吧  详情 回复 发表于 2024-1-24 21:32
 
 
 

回复

290

帖子

0

TA的资源

一粒金砂(高级)

板凳
 
LitchiCheng 发表于 2024-1-24 16:44 隔壁我拿的是RV1106,就很舒服了,256的nanoflash

羡慕啊,我试试SD卡启动应该会好一点吧

个人签名

阅读改变人生

 
 
 

回复
您需要登录后才可以回帖 登录 | 注册

随便看看
查找数据手册?

EEWorld Datasheet 技术支持

相关文章 更多>>
关闭
站长推荐上一条 1/6 下一条

 
EEWorld订阅号

 
EEWorld服务号

 
汽车开发圈

About Us 关于我们 客户服务 联系方式 器件索引 网站地图 最新更新 手机版

站点相关: 国产芯 安防电子 汽车电子 手机便携 工业控制 家用电子 医疗电子 测试测量 网络通信 物联网

北京市海淀区中关村大街18号B座15层1530室 电话:(010)82350740 邮编:100190

电子工程世界版权所有 京B2-20211791 京ICP备10001474号-1 电信业务审批[2006]字第258号函 京公网安备 11010802033920号 Copyright © 2005-2025 EEWORLD.com.cn, Inc. All rights reserved
快速回复 返回顶部 返回列表