使用python编写对图像加噪的代码
时间: 2024-02-03 22:04:06 浏览: 88
基于深度学习堆栈自动编码器模型的图像去噪python代码
可以使用Python的OpenCV库来实现对图像加噪,具体的代码如下:
```python
import cv2
import numpy as np
# 加载图像
img = cv2.imread('image.jpg')
# 添加高斯噪声
noise = np.zeros(img.shape, np.uint8)
cv2.randn(noise, 0, 50) # 均值为0,方差为50的高斯分布
img_noise = cv2.add(img, noise)
# 添加椒盐噪声
noise = np.zeros(img.shape, np.uint8)
cv2.randu(noise, 0, 255) # 生成0到255的随机数
img_noise[np.where(noise < 10)] = 0 # 10以下的随机数变为黑色
img_noise[np.where(noise > 245)] = 255 # 245以上的随机数变为白色
# 显示图像
cv2.imshow('Original', img)
cv2.imshow('Gaussian Noise', img_noise)
cv2.waitKey(0)
cv2.destroyAllWindows()
```
这段代码会加载一张名为`image.jpg`的图像,并分别添加高斯噪声和椒盐噪声,最后显示原图和加噪后的图像。可以根据需要调整噪声的参数和生成方式。
阅读全文