RK1808M0用yoloV5推理，结果与onnx模型预测严重不符

[复制链接] · 发表于 2021-7-14 17:33:02

工作需要，想用rk1808运行yoloV5目标检测模型。
1.网上下了代码，https://toscode.gitee.com/splendon/yolov5_for_rknn
2.使用上述代码将训练好的.pt模型》onnx》.rknn
3.将.rknn在 RK1808M0板子上运行，结果与中间步骤的onnx模型结果严重不符，其中onnx的结果已经确认正确了。

其中以上2步骤，在win10,ubantu,都转过，但这两者的预测结果不一样，也都不正确。
搞了几天了，也不知道问题再哪，现在不知道该从哪里下手了，技术支持大侠帮帮忙！谢谢了

只看该作者 · 发表于 2021-7-21 12:57:27

同问，onnx的结果我也没问题，到rknn结果就离谱

只看该作者 · 发表于 2021-9-17 15:18:17

yichicd 发表于 2021-7-21 12:57
同问，onnx的结果我也没问题，到rknn结果就离谱

是的，目标框乱，类别也乱

只看该作者 · 发表于 2022-4-18 19:18:19

各位老哥，问题解决了吗？