更改官方示例07_yolov5_sample时候出现问题

之前成功在pytorch训练了一版检测红绿灯的模型,现在想部署到X3pi上,已经成功通过工具链转换出bin文件了。想在07_yolov5_sample示例内替换bin文件和图片链接来检测模型的结果,已经按照B站视频内修改了coco_classes.names中的数据、postprocess.py后处理文件也修改过了,然后test_yolov5.py内的两个路径也都修改过了。但是运行结果还是报错,如下图所示

暂时不清楚原因。

对于coco_claseees.names、postprocess.py的修改如下所示:

介意发一下小玺玺大佬的那篇文章吗

哥们,解决了吗

你好,可以打断点看一下,三个分支数据的输出情况,然后对比一下自己的解码模块信息,基本就可以判断出来应该修改的数值

请问在那里查看三个分支数据的输出情况

官方示例的我暂时没有跑通,我跑通的是小玺玺大佬的那篇。照着直接一步步操作就可以

就在开发者论坛内:developer.horizon.ai/forumDetail/112555549341653639