下载官方代码对yolov5s.pt进行测试,出现如下报错:IndexError: index 57 is out of bounds for axis 0 with size 21
Traceback (most recent call last):
File "D:\1-syf\yolov5\val.py", line 359, in <module>
main(opt)
File "D:\1-syf\yolov5\val.py", line 333, in main
run(**vars(opt))
File "C:\ProgramData\Anaconda3\lib\site-packages\torch\autograd\grad_mode.py", line 27, in decorate_context
return func(*args, **kwargs)
File "D:\1-syf\yolov5\val.py", line 213, in run
confusion_matrix.process_batch(predn, labelsn)
File "D:\1-syf\yolov5\utils\metrics.py", line 156, in process_batch
self.matrix[detection_classes[m1[j]], gc] += 1 # correct
IndexError: index 57 is out of bounds for axis 0 with size 21
该错误是由于索引超出了列表的长度引起的,尝试输出predn和labelsn,但是发现也不会改。
隔壁有个和你一样的问题,两个号?
你用的是coco训练出来的模型,共80类,但是你验证的时候是voc20类,那么我coco预测出来类别id为57,你coco就越界了。
coco[57]是chair,对应voc[8]=chair。不信的话看下你的图片里是不是有椅子。
如果你要验证voc,你需要训练voc的模型而不是用coco训练出来的预训练模型
这个怎么解决啊
调用错模型了