morpological_operations.py 文件源码

def get_morphological_mask(point, omega, radius=5, shape='circle', morpho_patch=None):

    if morpho_patch is None:
        d = omega.ndim
        morpho_patch = get_morphological_patch(d, shape=shape)

    mask = np.zeros(omega, dtype=np.bool)
    mask.itemset(tuple(point), 1)
    for _ in range(radius):
        mask = ndimage.binary_dilation(mask, structure=morpho_patch).astype(mask.dtype)
    return mask