--------------------------------------------------------------------------- IndexError Traceback (most recent call last) <ipython-input-5-e3566b658d46> in <module> 41 # disable only the cluster number 2 42 masked_image = np.copy(image) ---> 43 masked_image[labels == 2] = [0, 0, 0] 44 45 # show the image IndexError: boolean index did not match indexed array along dimension 0; dimension is 469 but corresponding boolean dimension is 381297 #2