if image.shape != mask.shape:
        raise ValueError(
            "Image shape %s != mask shape %s" % (repr(image.shape), repr(mask.shape))