固定随机种子添加可复现高斯噪声后恢复原图的问题求助

问题原因 #

恢复结果不一致主要由三个错误导致：

• uint8类型的自动溢出截断：uint8类型的数值做加减运算时会自动对256取模，比如200 + 100 = 44、10 - 20 = 246，初始代码直接用uint8做加噪、减噪计算，得到的数值本身就不符合预期。
• 高斯噪声类型转换错误：高斯噪声包含负值，直接转uint8会把负值强制转换为255附近的正值，和预期的噪声分布完全不符。
• 图像读写的有损压缩：如果存储加噪图像时使用JPG等有损压缩格式，读写过程会丢失像素信息，导致读取的加噪图和实际计算得到的加噪图存在像素差异。

需修改的核心点 #

1. 所有涉及加减的计算环节统一转换为int32类型执行，仅在最终写入图像时转换为uint8
2. 高斯噪声不要直接转uint8，保持int32类型参与运算，避免负值被错误截断
3. 存储图像时选择PNG等无损压缩格式，读写时统一使用cv2.IMREAD_UNCHANGED参数，避免OpenCV自动修改通道数与位深
4. 溢出标记flag需要在int32计算阶段生成，确保标记的是真实超出0-255范围的位置

可行代码实现 #

加噪与标记生成 #

import numpy as np
import cv2

SEED = 1234
np.random.seed(SEED)

original = cv2.imread("C:/Users/user/project/1.png", cv2.IMREAD_UNCHANGED)
# 噪声转int32，避免负值截断
noise = np.random.normal(loc=0, scale=20, size=original.shape).astype(np.int32)

# 原图也转int32后再加噪
noisy_calc = original.astype(np.int32) + noise

floor_flag = np.where(noisy_calc < 0, 1, 0)
ceil_flag = np.where(noisy_calc > 255, 1, 0)
# 生成溢出标记：1表示加噪时溢出，恢复时直接保留加噪图像素
flag = floor_flag + ceil_flag

# 溢出位置保留原图像素，否则用加噪结果
noisy_img = np.where(flag == 0, noisy_calc, original).astype(np.uint8)
# 存为无损PNG
cv2.imwrite("C:/Users/user/project/1-1.png", noisy_img)

图像恢复 #

np.random.seed(SEED)
# 读取加噪图时用IMREAD_UNCHANGED，转int32参与计算
recover_noisy = cv2.imread("C:/Users/user/project/1-1.png", cv2.IMREAD_UNCHANGED).astype(np.int32)
# 生成和加噪阶段完全一致的噪声
noise = np.random.normal(loc=0, scale=20, size=recover_noisy.shape).astype(np.int32)

# 未溢出位置减噪声恢复，溢出位置直接保留
recovered_img = np.where(flag == 0, recover_noisy - noise, recover_noisy).astype(np.uint8)

# 验证恢复结果和原图完全一致
assert (recovered_img == original).all()

内容的提问来源于stack exchange，提问作者Jaden Park