当前位置: 代码迷 >> 综合 >> IndexError: boolean index did not match indexed array along dimension 0; dimension is 1 but correspo
  详细解决方案

IndexError: boolean index did not match indexed array along dimension 0; dimension is 1 but correspo

热度:44   发布时间:2024-01-05 11:34:59.0

IndexError: boolean index did not match indexed array along dimension 0; dimension is 1 but corresponding boolean dimension is 255

使用mask rcnn训练模型,但是没使用labelme标注工具,使用的是二值化图像,出现标题所示的情况。

搞了一下午,终于找到原因了!!!

我的数据集使用的是二值化图像,变为彩色图像的话,本来白色区域会变为0-255任意数值,所以可能会导致出现255中类别,但是我的数据集只有1个类别,因此,将只有一类别的所有不是背景的像素值变为相同值,就是只有1个类别了。!!!

不知道你们是什么原因,你们可以查看你们的mask,看是不是像素值个数与类别是否一样。

  相关解决方案