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模型是可以正常推理并输出推理结果的。
“该yolov5-lite.rknn模型在服务器上由自己训练的模型转化而来,使用python版rknn推理代码跑通了该模型并正确检测出了结果”,是指 PC 的仿真上跑通了吗? 那 板子 上的 python rknn-toolkit2 能跑通吗?只是用 C++ 库出问题?
如果是请检查 以下PC 上和板子上的库的版本一不一致 板蓝根 发表于 2022-11-23 10:15
“该yolov5-lite.rknn模型在服务器上由自己训练的模型转化而来,使用python版rknn推理代码跑通了该模型并正 ...
感谢,PC端使用的是1.4版本的toolkit2,现在将板子上的RKNPU驱动更新到1.4后能成功运行了。另外,关于你说的板子上跑python版的rknn toolkit2的说法,我注意到官方在对rknn toolkit2安装介绍时说该工具仅支持PC端,是确定板子上也能跑吗?只需要安装对应的python环境就可以了吗?
页:
[1]