Firefly开源社区

打印 上一主题 下一主题

[人工智能] rknn_run提示malloc错误

44

积分

0

威望

0

贡献

技术小白

积分
44

rknn_run提示malloc错误

发表于 2022-11-22 09:46:07      浏览:6187 | 回复:2        打印      只看该作者   [复制链接] 楼主
  • 固件类型: 官方提供的固件
  • 固件文件名称: ITX-3588J_Ubuntu20.04-Gnome-r21199_v1.0.1b_220812.
  • 固件下载地址: 官方下载地址
  • Log日志: none.zip
问题描述及复现步骤:
我在RK3588上编译运行yolov5-lite.rknn模型时,在rknn_run阶段报了内存错误,如附图所示。我使用新建的全1opencv矩阵代替了真实图片(省去图片预处理操作),直接将矩阵内存块复制到rknn_input,然后执行rknn_run。有大佬提供一些解决思路吗?非常感谢。

PS:1. 该yolov5-lite.rknn模型在服务器上由自己训练的模型转化而来,使用python版rknn推理代码跑通了该模型并正确检测出了结果,所以我认为模型本身应该是没有问题的吧;2. 附图展示的c++推理代码在原封不动的情况下跑yolov5s.rknn模型是可以正常推理并输出推理结果的。

malloc_error.png (76.41 KB, 下载次数: 534)

malloc_error.png

none.zip

150 Bytes, 下载次数: 0, 下载积分: 灯泡 -1 , 经验 -1

回复

使用道具 举报

1万

积分

14

威望

13

贡献

管理员

Rank: 9Rank: 9Rank: 9

积分
11182

优秀版主

发表于 2022-11-23 10:15:24        只看该作者  沙发
“该yolov5-lite.rknn模型在服务器上由自己训练的模型转化而来,使用python版rknn推理代码跑通了该模型并正确检测出了结果”,是指 PC 的仿真上跑通了吗? 那 板子 上的 python rknn-toolkit2 能跑通吗?只是用 C++ 库出问题?

如果是请检查 以下PC 上和板子上的库的版本一不一致
回复

使用道具 举报

44

积分

0

威望

0

贡献

技术小白

积分
44
发表于 2022-11-30 09:35:08        只看该作者  板凳
板蓝根 发表于 2022-11-23 10:15
“该yolov5-lite.rknn模型在服务器上由自己训练的模型转化而来,使用python版rknn推理代码跑通了该模型并正 ...

感谢,PC端使用的是1.4版本的toolkit2,现在将板子上的RKNPU驱动更新到1.4后能成功运行了。另外,关于你说的板子上跑python版的rknn toolkit2的说法,我注意到官方在对rknn toolkit2安装介绍时说该工具仅支持PC端,是确定板子上也能跑吗?只需要安装对应的python环境就可以了吗?
回复

使用道具 举报

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

本版积分规则

友情链接 : 爱板网 电子发烧友论坛 云汉电子社区 粤ICP备14022046号-2
快速回复 返回顶部 返回列表