Firefly开源社区

标题: rk3588 rknpu 上板推理检测框不对 [打印本页]

作者: Genos 时间: 2024-6-3 15:11
标题: rk3588 rknpu 上板推理检测框不对
本帖最后由 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, 下载次数: 224)

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

作者: cccmm 时间: 2024-7-1 11:26
我也碰到了一样的问题，请问有找到解决办法吗，跪谢

作者: dengkx 时间: 2024-7-2 15:34

cccmm 发表于 2024-7-1 11:26
我也碰到了一样的问题，请问有找到解决办法吗，跪谢

这看着像是图像分辨率与模型分辨率不同导致的

欢迎光临 Firefly开源社区 (https://dev.t-firefly.com/)