给图片加上高斯噪声

原创 python opencv

在有些任务中，需要在原数据集上加入噪声，下面程序演示了将 path 中的图片加上高斯噪声然后保存到 path2 中

通过 skimage-lib 给图片加上高斯噪声

import matplotlib.pyplot as plt
from skimage import util
from skimage import io
import numpy as np
import os


path = "/data4/wangyh/LIR-for-Unsupervised-IR/VOCdevkit/VOC2012/JPEGImages/"  # 文件路径
path2 = "/data4/wangyh/LIR-for-Unsupervised-IR/VOCdevkit/VOC2012/TJPEGImages/"
for file in os.listdir(path):
    img = io.imread(path+file)
    noise_gs_img = util.random_noise(img,mode='gaussian') # gaussian 高斯加性噪声
    max = noise_gs_img.max()
    noise_gs_img = noise_gs_img*255/max
    noise_gs_img = np.uint8(noise_gs_img)
    io.imsave(path2+file, noise_gs_img)
    print(file, 'OK')

'''
其他噪声
noise_salt_img = util.random_noise(img,mode='salt')  # 盐噪声，随机用1替换像素。属于高灰度噪声
noise_pepper_img = util.random_noise(img,mode='pepper')  # 胡椒噪声，随机用0或-1替换像素，属于低灰度噪声
noise_sp_img = util.random_noise(img,mode='s&p')  # 椒盐噪声，两种噪声同时出现，呈现出黑白杂点
noise_speckle_img = util.random_noise(img,mode='speckle')  # out = image + n *图像的乘法噪声，其中n是具有指定均值和方差的均匀噪声
'''

本文由 Yonghui Wang 创作，采用 知识共享署名4.0 国际许可协议进行许可
本站文章除注明转载/出处外，均为本站原创或翻译，转载前请务必署名
最后编辑时间为: Dec 19, 2024 12:13 pm