tools.py 文件源码

def add_noise(x):
    """Add random noise to images."""
    sz = x.shape
    mask = (x == 0) * 1.
    noise = np.random.rand(sz[0], sz[1])
    for i in range(x.shape[0]):
        noise[i] = ndimage.uniform_filter(noise[i].reshape(28, 28), size=3).flatten()
    noise = np.multiply(noise, mask)
    out = np.clip(x + noise, 0., 1.)
    return out