Python图片添加噪声

import skimage
import os
from skimage import io
import random



def addGaussNoise(s):
    var = random.uniform(0.0001, 0.04)
    origin = skimage.io.imread(s)
    noisy = skimage.util.random_noise(origin, mode=\'gaussian\', var=var)
    return noisy


def addSaltNoise(s):
    var = random.uniform(0.01, 0.2)
    origin = skimage.io.imread(s)
    noisy = skimage.util.random_noise(origin, mode=\'s&p\', amount=var)
    return noisy


def addSpeckleNoise(s):
    var = random.uniform(0.0001, 0.04)
    origin = skimage.io.imread(s)
    noisy = skimage.util.random_noise(origin, mode=\'speckle\', var=var)
    return noisy


if __name__ == \'__main__\':
    file_pathname = \"./data/jpg/\"
    save_pathname = \"./data/demo2/\"
    for filename in os.listdir(file_pathname):
        index = random.randint(0, 3)
        if index == 0:  # 如果是0，什么都不做
            continue
        elif index == 1:  # 添加高斯噪声
            res = addGaussNoise(file_pathname + \'/\' + filename)
            io.imsave(save_pathname + filename, res)
        elif index == 2:  # 添加椒盐噪声
            res = addGaussNoise(file_pathname + \'/\' + filename)
            io.imsave(save_pathname + filename, res)
        elif index == 3:  # 添加乘法噪声
            res = addGaussNoise(file_pathname + \'/\' + filename)
            io.imsave(save_pathname + filename, res)

如果有任何问题可以邮箱联系我:raymondlam1@yeah.net