mask-rcnn报错: IndexError: boolean index did not match indexed array along dimension 0; ......

mask-rcnn在执行train.py训练模型时,报错如下:

image_id 769
ERROR:root:Error processing image {'id': 769, 'source': 'shapes', 'path': 'train_data/pic/4A707SG3.jpg', 'width': 820, 'height': 706, 'mask_path': 'train_data/cv2_mask/4A707SG3.png', 'yaml_path': 'train_data/labelme_json/4A707SG3_json/info.yaml'}
Traceback (most recent call last):
  File "D:\python-workspace\Mask_RCNN-master2\mrcnn\model.py", line 1712, in data_generator
    use_mini_mask=config.USE_MINI_MASK)
  File "D:\python-workspace\Mask_RCNN-master2\mrcnn\model.py", line 1268, in load_image_gt
    class_ids = class_ids[_idx]
IndexError: boolean index did not match indexed array along dimension 0; dimension is 0 but corresponding boolean dimension is 1

mask-rcnn报错: IndexError: boolean index did not match indexed array along dimension 0; ......_第1张图片

 

问题定位:

mask-rcnn报错: IndexError: boolean index did not match indexed array along dimension 0; ......_第2张图片

 

你可能感兴趣的:(目标检测,mask_rcnn,目标检测,机器学习)