1. 首页首页
  2. 编程日记编程日记

rknn（rknn模型）

编程日记 • • 阅读 66

rknn（rknn模型）感兴趣的可以关注一下公众号 会第一时间给您推送更多精彩的内容 欢迎大家前来指正 欢迎欢迎 官方给的一个流程 RKNN 模型推理测试为了避免踩坑在开头提出来 按照官方的流程可以跑通 他自己提供的 yolov5s rknn 640 640 的模型 但是跑自己的模型的时候加载就会出错 E RKNN failed to decode config data Segmentation fault



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

官方给的一个流程: 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_45715405/article/details/build-linux.sh -t rv1106 -a armv7l -d yolov5
 

 

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

 
 

 

测试main.cpp
 

 

touch CMakeLists.txt
 

 

测试使用
 


                    


                    


                    

                        

                            

                                编程小号
                            

                            

                                

                                    
                                        
                                    
                                

                            

                        

                    

                


                

                    
                    
                    

                        
                        

                             上一篇
                            2026-02-02 22:21
                        

                    

                    
                    


                    
                    
                    

                        
                        

                            下一篇 
                            2026-02-02 23:01
                        

                    

                    
                    
                


                

                    
相关推荐

                    
                

                

                    

                        版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容，
                        请发送邮件至 举报，一经查实，本站将立刻删除。
如需转载请保留出处：https://bianchenghao.cn/bian-cheng-ri-ji/39420.html