此文記錄自己在做圖像數(shù)據(jù)增強(qiáng)給圖像加噪時(shí)遇到的一些情況。

椒鹽噪音：簡(jiǎn)單點(diǎn)說(shuō)就是在圖像中隨機(jī)加一些白點(diǎn)或者黑點(diǎn)。

from skimage import io

import random

import numpy as np

def salt_and_pepper_noise(img, proportion=0.05):

noise_img =img

height,width =noise_img.shape[0],noise_img.shape[1]

num = int(height*width*proportion)#多少個(gè)像素點(diǎn)添加椒鹽噪聲

for i in range(num):

w = random.randint(0,width-1)

h = random.randint(0,height-1)

if random.randint(0,1) ==0:

noise_img[h,w] =0

else:

noise_img[h,w] = 255

return noise_img

if __name__ == '__main__':

img =io.imread(r"C:\work\data\he\guangtou\0000709.jpg")

noise_img = salt_and_pepper_noise(img)

io.imshow(noise_img)

io.show()

效果圖：

高斯噪音：給每一個(gè)像素點(diǎn)加一個(gè)隨機(jī)的符合高斯分布的值。

from skimage import io

import random

import numpy as np

def gauss_noise(image):

img = image.astype(np.int16)#此步是為了避免像素點(diǎn)小于0，大于255的情況

mu =0

sigma = 10

for i in range(img.shape[0]):

for j in range(img.shape[1]):

for k in range(img.shape[2]):

img[i,j,k] = img[i,j,k] +random.gauss(mu=mu,sigma=sigma)

img[img>255] = 255

img[img<0] = 0

img = img.astype(np.uint8)

return img

if __name__ == '__main__':

img =io.imread(r"C:\work\data\he\guangtou\0000709.jpg")

# noise_img = salt_and_pepper_noise(img)

gauss_img = gauss_noise(img)

io.imshow(gauss_img)

io.show()

總結(jié)

以上是生活随笔為你收集整理的imgaug批量椒盐噪声 python_python手写给图像加椒盐噪声和高斯噪声的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。