使用val.py测试yolov5s.pt报错:IndexError: index 57 is out of bounds for axis 0 with size 21

问题遇到的现象和发生背景

下载官方代码对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训练出来的预训练模型

这个怎么解决啊

调用错模型了