Traceback (most recent call last):
File "train.py", line 542, in
train() # train normally
File "train.py", line 418, in train
save_json=final_epoch and epoch > 0 and 'coco.data' in data)
File "/home/sdu/yolov3-channel-and-layer-pruning/test.py", line 194, in test
maps[c] = ap[i]
IndexError: index 2 is out of bounds for axis 0 with size 2
,其他分类训练没有问题,只要classes是2,就报上面的错误,是怎么回事
Traceback (most recent call last): File "train.py", line 542, in
train() # train normally
File "train.py", line 418, in train
save_json=final_epoch and epoch > 0 and 'coco.data' in data)
File "/home/sdu/yolov3-channel-and-layer-pruning/test.py", line 194, in test
maps[c] = ap[i]
IndexError: index 2 is out of bounds for axis 0 with size 2
,其他分类训练没有问题,只要classes是2,就报上面的错误,是怎么回事