RDK X3部署yolov5出错

一个好的问题会收获一个好的答案,提问前请参考提问模板提问:

标题需要包含基本的信息,请以[问题分类][板卡类型][问题简述]的形式填写标题。当一个问题符合多个标签时,也可以提供多个标签。标题举例:[图像算法][RDK X5][xx案例运行现象不符合预期]

[问题分类]参考:[多媒体][图像算法][TROS][语音算法][强化学习][大模型][激光雷达][算法工具链][Linux开发][RDK Studio][RDK Model Zoo][GitHub][… 其他 …]
[板卡类型]参考:[RDK X3][RDK X3 Module][RDK X5][RDK X5 Module][RDK Ultra][RDK S100][… 其他 …]
针对特定的问题,请参考特定领域问题的发帖模板继续提供相关信息: forum.d-robotics.cc/t/topic/28613
前提说明
版本:RDK X3 ros2 humble
模型训练:YOLOV5 v7.0训练,修改导出具有三输出的onnx
模型转换:使用工具链输出bin文件,使用04_inference.sh 一切正常

使用算法工具链从 bash 01_check_X3.sh 到 bash 04_inference.sh 一切正常:


接下来,使用python板卡部署的时候,出现检测框乱排列的现象

这是修改的部分postprocess.py

尝试使用rdk_model_zoo:

这是bin模型可视化:

您好,使用OE示例包,bash 04_inference.sh 推理正常说明本身模型量化过程没有问题,但是由于整体模型量化过程不是根据ModelZoo中的流程来走的,所以此模型的后处理无法与Model Zoo中的代码对应,需要参考04_inference.sh 中调用的后处理代码对板端推理python代码的后处理进行修改适配。