rknn（rknn模型）



感兴趣的可以关注一下公众号，会第一时间给您推送更多精彩的内容，欢迎大家前来指正，欢迎欢迎~~

官方给的一个流程: RKNN 模型推理测试为了避免踩坑在开头提出来

按照官方的流程可以跑通，他自己提供的yolov5s.rknn（640*640）的模型，但是跑自己的模型的时候加载就会出错E RKNN: failed to decode config data!Segmentation fault (core dumped)，应该是这个地址的链接版本太老了，并且给出来的demo预处理也没有使用librga做硬件加速，直接跳过，官方也给出来了说这个github链接不在维护（https://github.com/rockchip-linux/rknpu2）

官方最新地址: rknn-toolkit2
所有实例程序就都在这里了
工具包库: rknn_model_zoo
下载下来

交叉编译需要先安装 LuckFox Pico SDK

在 rknn_model_zoo目录下我们现在对这些例程进行交叉编译，编译例程前需要设置如下环境变量：

cmake要求必须大于3.15，不满足自行百度解决

编译完成之后会在当前目录下生成一个install目录进入里面就可以找到编译出来的程序和库和模型文件

通过adb，将库和model下的文件文件都推上去

可以看到没有任何问题，但是这就是最大的问题

如果直接拿自己的模型不修改代码，直接推理，会出现这种报错，就是类别没有对上

从原本80修改成实际类别数，还有板上model目录下的coco_80_labels_list.txt文件也需要修改成你的类别名

进行推理

推理出来的结果就是这样子了，矩形框都是错乱的

这个文件就是源码里面埋了很大一个坑，取反了，模型输入要求是NHWC，但是他将h取到w上面了，所以出现错乱，为什么官方自己的模型没问题推理是正常的，因为官方的是640*640，高宽尺寸是一致的，所以没有出现这个bug

修改之后再次验证正常了

将示例程序封装成so动态库，嵌入到其他地方使用

cmake文件，任使用官方给的脚本进行调用https://blog.csdn.net/weixin_/article/details/build-linux.sh -t rv1106 -a armv7l -d yolov5

编写cpp进行调用封装的库，进行推理cmake文件

测试main.cpp

touch CMakeLists.txt

测试使用

版权声明：

本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如若内容造成侵权、违法违规、事实不符，请将相关资料发送至xkadmin@xkablog.com进行投诉反馈，一经查实，立即处理！

转载请注明出处，原文链接：https://www.xkablog.com/rfx/19701.html