小波变换图像去噪python

以下是使用Python进行小波变换图像去噪的示例代码：

import pywt
import cv2

# 读取图像
img = cv2.imread('image.jpg', 0)

# 进行小波变换
coeffs = pywt.dwt2(img, 'haar')
cA, (cH, cV, cD) = coeffs

# 设置阈值
threshold = 30

# 对小波系数进行去噪
cA = pywt.threshold(cA, threshold)
cH = pywt.threshold(cH, threshold)
cV = pywt.threshold(cV, threshold)
cD = pywt.threshold(cD, threshold)

# 重构图像
img_denoised = pywt.idwt2((cA, (cH, cV, cD)), 'haar')

# 显示图像
cv2.imshow('Original Image', img)
cv2.imshow('Denoised Image', img_denoised)
cv2.waitKey(0)
cv2.destroyAllWindows()

在代码中，首先读取图像，并对其进行小波变换。然后，设置阈值并对小波系数进行去噪。最后，使用逆小波变换重构图像，并显示原始图像和去噪后的图像。可以根据需要调整阈值以达到更好的去噪效果。