binary_dilation_erosion.py 文件源码

def erosion(img):
    """Perform Erosion on an image.
    Args:
        img The image to be changed.
    Returns:
        Changed image
    """
    img = np.copy(img).astype(np.float32)
    mask = np.array([[1, 1, 1], [1, 1, 1], [1, 1, 1]])
    corr = signal.correlate(img, mask, mode="same") / 9.0
    corr[corr < 255.0 - 1e-8] = 0
    corr = corr.astype(np.uint8)
    return corr