Firefly开源社区

打印 上一主题 下一主题

[应用开发] rk3588 rknpu 上板推理检测框不对

14

积分

0

威望

0

贡献

技术小白

积分
14

rk3588 rknpu 上板推理检测框不对

发表于 2024-6-3 15:11:44      浏览:74 | 回复:0        打印      只看该作者   [复制链接] 楼主
本帖最后由 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



图1.jpg (140.63 KB, 下载次数: 0)

图1.jpg

图2.jpg (165.61 KB, 下载次数: 0)

图2.jpg
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

友情链接 : 爱板网 电子发烧友论坛 云汉电子社区 粤ICP备14022046号-2
快速回复 返回顶部 返回列表