python 离散小波变换函数的相关应用

import cv2
from matplotlib import pyplot as plt
import matplotlib.image as mpimg
import matplotlib as mpl
#mpl.rcParams['figure.dpi'] = 300
import pywt
import pywt.data
import numpy as np
from math import sqrt
from PIL import Image, ImageChops, ImageOps
import pickle


#加载图像
img = cv2.imread('img/peter-3794597_1280.jpeg')
image = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
# plt.imshow(image,cmap='gray');
# plt.show()

#小波变换
c = pywt.dwt2(image,wavelet)
cA, (cH, cV, cD)= c

#显示cA cH cV cD
titles = ['cA', 'cH', 'cV', 'cD']
fig = plt.figure(figsize=(12, 3))
for i, a in enumerate([cA, cH, cV, cD]):
  ax = fig.add_subplot(1, 4, i + 1)
  ax.imshow(abs(a), interpolation="nearest", cmap = 'gray')
  ax.set_title(titles[i], fontsize=20)
fig.tight_layout()
plt.show()
#逆变换
w, h = image.shape  
c_inverse = pywt.idwt2((c),wavelet)[:h,:w]
titles = ['原图像', '恢复后图像']
fig = plt.figure(figsize=(10, 6))
for i, a in enumerate([image,c_inverse]):
  ax = fig.add_subplot(1, 2, i + 1)
  ax.imshow(a, interpolation="nearest", cmap = 'gray')
  ax.set_title(titles[i], fontsize=15)
fig.tight_layout()
plt.show()

#确定像素值值位零的点的个数
print((sum(sum(sum(c[1][m][n] == 0) for n in range(len(c[1][0]))) for m in range(len(c[1])))) + sum(sum(c[0][m] == 0) for m in range(len(c[0]))))
#设定阀值
threshold = 10
#阀值以下灰度值去零
h1 = cA.shape[0] 
for i in range(h1):
      c[0][i][np.abs(c[0][i])<threshold] = 0.0 
      for j in range(len(c[1])):
         c[1][j][i][np.abs(c[1][j][i])<threshold] = 0.0 
print('零点 =',(sum(sum(sum(c[1][m][n] == 0) for n in range(h1)) for m in range(len(c[1])))) + sum(sum(c[0][m] == 0) for m in range(h1)))