我在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)
-
150 Bytes, 下载次数: 0, 下载积分: 灯泡 -1 , 经验 -1