geometry.py 文件源码

def gaussian(sigma=0.5, samples=128):
    ''' Generates a gaussian mask with a given sigma

    Args:
        sigma (`float`): width parameter of the gaussian, expressed in radii of
            the output array.

        samples (`int`): number of samples in square array.

    Returns:
        `numpy.ndarray`: mask with gaussian shape.

    '''
    s = sigma

    x = np.arange(0, samples, 1, float)
    y = x[:, np.newaxis]

    # // is floor division in python
    x0 = y0 = samples // 2
    return exp(-4 * log(2) * ((x - x0 ** 2) + (y - y0) ** 2) / (s * samples) ** 2)