preprocessor.py 文件源码

def random_noise(img):  #TODO dataug
    """
    Puts random noise on image
    :param img: image without noise
    :return: image with noise
    """
    severity = np.random.uniform(0, 0.6)
    blur = ndimage.gaussian_filter(np.random.randn(*img.shape) * severity, 1)
    img_speck = (img + blur)
    img_speck[img_speck > 1] = 1
    img_speck[img_speck <= 0] = 0

    return img_speck