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,看是不是像素值个数与类别是否一样。