IndexError: index 13 is out of bounds for axis 0 with size 4

[piscesdream]

hello! 为什莫会出现这样的错误？我的是3分类，为什莫index会出现13。
Traceback (most recent call last):
File "test.py", line 397, in
test(opt.data,
File "test.py", line 236, in test
confusion_matrix.process_batch(predn, torch.cat((labels[:, 0:1], tbox), 1))
File "/root/yolo/yolov7_plate/utils/metrics.py", line 148, in process_batch
self.matrix[detection_classes[m1[j]], gc] += 1 # correct
IndexError: index 13 is out of bounds for axis 0 with size 4

[we0091234]

是不是最后一个epoch出现的

[piscesdream]

单独跑test.py这个脚本出现的，训练时是最后出现的

[we0091234]

单独跑test.py这个脚本出现的，训练时是最后出现的

test.py 我还没改，你是要跑指标，结果吗？

[piscesdream]

对，目前在绘制混淆矩阵会出现这个错误，其他暂无影响

[we0091234]

对，目前在绘制混淆矩阵会出现这个错误，其他暂无影响

好吧，我找个时间看看吧，最近挺忙的

[we0091234]

我改了一下，可以跑通，但是我两类，输出只有一类，奇怪

[we0091234]

对，目前在绘制混淆矩阵会出现这个错误，其他暂无影响

问题解决了，你重新拉取一下代码就可以了