python PIL图像随机添加噪声

原创

mob64ca12d68df5 2024-10-25 06:38:04 ©著作权

文章标签 图像处理 Image Python 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob64ca12d68df5的原创作品，请联系作者获取转载授权，否则将追究法律责任

Python PIL图像随机添加噪声

随着图像处理技术的发展，图像的增强与去噪已经成为了计算机视觉领域中的重要话题。为此，学习如何向图像中添加随机噪声是一个值得掌握的技能。本文将采用Python中的Pillow库（PIL）来实现图像添加噪声的功能，并为您详细讲解实现的过程和原理。

噪声的种类

在图像处理中，常见的噪声有多种类型，包括但不限于：

高斯噪声（Gaussian Noise）：均值为0，方差为σ²的噪声，呈现正态分布。
椒盐噪声（Salt-and-Pepper Noise）：图像中随机出现黑白点，模拟数字传输错误。
泊松噪声（Poisson Noise）：与光子统计有关的噪声，适用于光学成像。

在本示例中，我们将实现添加高斯噪声和椒盐噪声的功能。

安装相关库

请确保您已安装Pillow库。如果尚未安装，可以使用以下命令：

pip install Pillow

添加噪声的核心代码

下面是实现随机添加高斯噪声和椒盐噪声的Python代码：

import numpy as np
from PIL import Image
import random

class NoiseAdder:
    def __init__(self, image_path):
        self.image = Image.open(image_path)

    def add_gaussian_noise(self, mean=0, std=25):
        """添加高斯噪声"""
        img_array = np.array(self.image)
        gaussian_noise = np.random.normal(mean, std, img_array.shape)
        noisy_image = img_array + gaussian_noise
        noisy_image = np.clip(noisy_image, 0, 255).astype(np.uint8)
        return Image.fromarray(noisy_image)

    def add_salt_and_pepper_noise(self, salt_prob=0.02, pepper_prob=0.02):
        """添加椒盐噪声"""
        img_array = np.array(self.image)
        total_pixels = img_array.size
        num_salt = np.ceil(salt_prob * total_pixels)
        num_pepper = np.ceil(pepper_prob * total_pixels)

        # 添加盐噪声
        coords = [np.random.randint(0, i - 1, int(num_salt)) for i in img_array.shape]
        img_array[coords[0], coords[1], :] = 255

        # 添加椒噪声
        coords = [np.random.randint(0, i - 1, int(num_pepper)) for i in img_array.shape]
        img_array[coords[0], coords[1], :] = 0

        return Image.fromarray(img_array)

# 使用示例
if __name__ == "__main__":
    noise_adder = NoiseAdder("your_image_path.jpg")
    noisy_gaussian_image = noise_adder.add_gaussian_noise()
    noisy_pepper_salt_image = noise_adder.add_salt_and_pepper_noise()

    noisy_gaussian_image.show()
    noisy_pepper_salt_image.show()

代码解析

导入库：我们首先导入NumPy和Pillow库。
NoiseAdder类：该类的初始化方法接收图像路径并加载图像。
添加高斯噪声：通过生成高斯分布的随机数将噪声添加到图像上，并使用numpy的clip函数限制像素值在0-255之间。
添加椒盐噪声：计算盐和椒的数量，随机选择像素并将其值修改为黑白，分别代表椒盐噪声。

类图

以下是NoiseAdder类的类图：

classDiagram
    class NoiseAdder {
        +__init__(image_path: str)
        +add_gaussian_noise(mean: float, std: float)
        +add_salt_and_pepper_noise(salt_prob: float, pepper_prob: float)
    }

ER图

以下是与图像处理相关的实体关系图：

erDiagram
    IMAGE {
        STRING image_path
        INTEGER width
        INTEGER height
    }
    NOISE {
        STRING type
        FLOAT mean
        FLOAT std
    }
    IMAGE ||--o{ NOISE : "applies"