python 显示图像各级小波变换
时间: 2023-11-08 16:06:09 浏览: 182
图像的小波变换代码,简洁有效
4星 · 用户满意度95%
要显示图像的各级小波变换,可以使用 PyWavelets 库。以下是一个简单的示例代码:
```python
import pywt
import numpy as np
import matplotlib.pyplot as plt
# 读取图像
img = plt.imread('image.jpg')
# 将图像转换为灰度图
img_gray = np.mean(img, axis=2)
# 小波变换
coeffs = pywt.wavedec2(img_gray, 'haar', level=3)
# 显示各级小波系数
fig, axs = plt.subplots(2, 2)
axs[0, 0].imshow(coeffs[0], cmap='gray')
axs[0, 1].imshow(coeffs[1][0], cmap='gray')
axs[1, 0].imshow(coeffs[1][1][0], cmap='gray')
axs[1, 1].imshow(coeffs[1][1][1], cmap='gray')
plt.show()
```
这个示例代码将图像进行了三级小波变换,并显示了各级小波系数。其中,`coeffs` 表示第一级低频系数,`coeffs` 表示第二级低频系数,`coeffs` 和 `coeffs[1]` 分别表示第二级高频系数的水平和垂直方向,以及对角线方向。
阅读全文