|
【应用开发】
rk3588 rknpu 上板推理检测框不对
发表于 2024-6-3 15:11:44
浏览:1279
|
回复: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
|
|