前言
需要增加数据集,对图像进行增广
代码
from skimage.io import imread
from skimage import img_as_float
from skimage.util import random_noise
import matplotlib.pylab as plt
import os
if __name__ == '__main__':
images_dir = "D:/pycodes/python_image_processing/9781789343731_Code/images"
file_name = "parrot.png"
file_path = os.path.join(images_dir,file_name)
im = img_as_float(imread(file_path))
# 添加高斯噪声
plt.figure(figsize=(15,12))
sigmas = [0.1,0.25,0.5,1]
for i in range(4):
noisy = random_noise(im,var=sigmas[i]**2)
plt.subplot(2,2,i+1),plt.imshow(noisy),plt.axis('off')
plt.title(f'gaussian noise with sigma={str(sigmas[i])}',size=20)
plt.show()