roi_helpers.py报错

[yezhaoguli]

File "train.py", line 207, in <module>
    fit_one_epoch(model_rpn, model_all, epoch, epoch_size, epoch_size_val, gen, gen_val, Interval_Epoch, callback)
  File "train.py", line 55, in fit_one_epoch
    X2, Y1, Y2 = calc_iou(R, config, boxes[i], NUM_CLASSES)
  File "/media/user/date/qianyi/deeplearningwork/faster-rcnn-keras/utils/roi_helpers.py", line 110, in calc_iou
    Y1 = np.eye(num_classes)[np.array(gt_roi_label,np.int32)]
IndexError: index 6 is out of bounds for axis 0 with size 2

NUM_CLASSES=2，只有一个类可以进行训练吗出错如上，求助，谢谢

[bubbliiiing]

voc_annotation没改。

[yezhaoguli]

谢谢博主