yolov5.5.0训练模型时报错IndexError: index 1 is out of bounds for axis 0 with size 1如何解决
该问题主要问题就是数据集的实际标签类别数和lables文件夹中的txt文件的索引类别不同。例如实际标签类别只有一种,而txt文件中索引出现了大于0的索引,因此就会数组越界。对于上述问题可以重新给数据集进行标记,或者通过循环读取txt文件列表,并将其中的类别索引修改。对于我的数据集进行训练过程中,发现了很多问题,例如上图报错。
原创
2023-07-12 20:48:51 ·
6437 阅读 ·
2 评论