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

Question

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

下载官方代码对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，但是发现也不会改。

Answer 1

隔壁有个和你一样的问题，两个号？
你用的是coco训练出来的模型，共80类，但是你验证的时候是voc20类，那么我coco预测出来类别id为57，你coco就越界了。
coco[57]是chair，对应voc[8]=chair。不信的话看下你的图片里是不是有椅子。
如果你要验证voc，你需要训练voc的模型而不是用coco训练出来的预训练模型