本文主要是介绍mask_rcnn:boolean index did not match indexed array along dimension 0; dimension is 0 ...(已解决),希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
问题:IndexError: boolean index did not match indexed array along dimension 0; dimension is 0 but corresponding boolean dimension is 1
在用mask_rcnn做目标检测,训练模型时,报错如下图所示,model.py里的
_idx = np.sum(mask, axis=(0, 1)) > 0
mask = mask[:, :, _idx]
class_ids = class_ids[_idx]
是正确的;错误的主要原因是train.py里面的分类classes列表出问题了,通常是在更换本地的数据集后,分类没有做修改,或者分类不完整导致。
解决方法:
或
这篇关于mask_rcnn:boolean index did not match indexed array along dimension 0; dimension is 0 ...(已解决)的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!