Firefly开源社区

标题: rknn_run提示malloc错误 [打印本页]

作者: Meeeteor 时间: 2022-11-22 09:46
标题: rknn_run提示malloc错误
我在RK3588上编译运行yolov5-lite.rknn模型时，在rknn_run阶段报了内存错误，如附图所示。我使用新建的全1opencv矩阵代替了真实图片（省去图片预处理操作），直接将矩阵内存块复制到rknn_input，然后执行rknn_run。有大佬提供一些解决思路吗？非常感谢。

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

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

none.zip

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

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

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

作者: Meeeteor 时间: 2022-11-30 09:35

板蓝根发表于 2022-11-23 10:15
“该yolov5-lite.rknn模型在服务器上由自己训练的模型转化而来，使用python版rknn推理代码跑通了该模型并正 ...

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

欢迎光临 Firefly开源社区 (https://dev.t-firefly.com/)