rknn_run提示malloc错误

Meeeteor

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

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

板蓝根

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

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

Meeeteor

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

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