|
|
【应用开发】
rk3588 rknpu 上板推理检测框不对
发表于 2024-6-3 15:11:44
浏览：1319
|
回复：2
打印
只看该作者
[复制链接]
楼主
本帖最后由 Genos 于 2024-6-4 09:10 编辑
$$正常部分
在PC端使用yolov5s coco128训练并检测best.pt，效果还可以,如图1所示
采用export.py导出best.onnx文件，output出三个节点。【1,255,80,80】【1,255,40,40】【1,255,20,20】
使用rknn-toolkit2-1.4.0/../yolov5中test.py，成功的将onnx转换成rknn，并进行adb连板调试npu，效果还可以，如图1所示。
$$错误部分
将rknn模型放入rknpu2_1.4.0/../rknn_yolov5_demo/model/RK3588中，并修改类别名称、类别数。成功生成build、install文件夹。
将install里文件放入rk3588板卡中，并运行./rknn_yolov5_demo，结果图2所示。
（其中，NMS_THRESH = 0.15，sigmoid函数已去除）
$$尝试
将rknpu2_1.4.0放入rk3588本地进行编译，结果与图2一致。
$$尝试
使用他人生成的rknn模型重新生成应用程序，结果与图2差不多（框多框杂）。
$$尝试
adb联板调试时，检测npu利用率，确实有调用。
$$尝试
直接跑官方内置的demo可以正确识别。
$$尝试
在板卡上编译rknn_toolkit_lite2，可以正确识别。
想咨询一下大佬有没有好办法?
跪谢！！！
板子信息补充：sdk version:1.4.0 driver version:0.8.2
|
|
