Toybrick

标题: RK1808M0用yoloV5推理，结果与onnx模型预测严重不符 [打印本页]

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

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

作者: yichicd 时间: 2021-7-21 12:57
同问，onnx的结果我也没问题，到rknn结果就离谱

作者: 慕容紫 时间: 2021-9-17 15:18

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

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

作者: wokafu 时间: 2022-4-18 19:18
各位老哥，问题解决了吗？

欢迎光临 Toybrick (https://t.rock-chips.com/)