如何在图像中添加赖斯噪声？

import cv2
import numpy as np

def add_rice_noise(image, K=1.5, scale=10):
    """
    在图像中添加赖斯噪声。
    
    :param image: 输入图像（numpy数组）
    :param K: 赖斯分布的参数K
    :param scale: 噪声的标准差
    :return: 添加噪声后的图像
    """
    h, w = image.shape[:2]
    noise = np.zeros_like(image)
    
    for i in range(h):
        for j in range(w):
            A = np.random.rayleigh(scale)  # 瑞利分布的幅度
            theta = np.random.uniform(0, 2 * np.pi)  # 均匀分布的相位
            noise[i, j] = A * np.exp(1j * theta)
    
    noisy_image = image + noise.real * K
    noisy_image = np.clip(noisy_image, 0, 255).astype(np.uint8)
    
    return noisy_image

# 读取图像
image = cv2.imread('input.jpg', cv2.IMREAD_GRAYSCALE)

# 添加赖斯噪声
noisy_image = add_rice_noise(image, K=1.5, scale=10)

# 显示结果
cv2.imshow('Original Image', image)
cv2.imshow('Noisy Image', noisy_image)
cv2.waitKey(0)
cv2.destroyAllWindows()