morpological_operations.py 文件源码

def get_morphological_patch(dimension, shape):
    """
    :param dimension: dimension of the image (NOT the shape).
    :param shape: circle or square.
    :return: morphological patch as ndimage
    """
    if shape == 'circle':
        morpho_patch = ndimage.generate_binary_structure(dimension, 1)
    elif shape == 'square':
        morpho_patch = ndimage.generate_binary_structure(dimension, 3)
    else:
        return IOError

    return morpho_patch