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

原创于 2022-02-14 11:32:11 发布 · 4.1k 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#目标检测 #机器学习

目标检测同时被 2 个专栏收录

25 篇文章

订阅专栏

mask_rcnn

12 篇文章

订阅专栏

在尝试使用Mask R-CNN模型进行训练时，遇到一个错误。具体表现为在data_generator函数中，因boolean索引与被索引数组维度不匹配导致IndexError。错误出现在尝试访问class_ids数组的特定元素时。问题可能源于数据预处理或输入数据的格式不正确。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

问题定位：