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