pywt

1
import matplotlib.pyplot as plt
2
import numpy as np
3

4
import pywt
5
import pywt.data
6

7
# Load image
8
original = pywt.data.camera()
9

10
# Wavelet transform of image, and plot approximation and details
11
titles = ['Approximation', ' Horizontal detail',
12
          'Vertical detail', 'Diagonal detail']
13
coeffs2 = pywt.dwt2(original, 'bior1.3')
14
LL, (LH, HL, HH) = coeffs2
15
fig = plt.figure(figsize=(12, 3))
16
for i, a in enumerate([LL, LH, HL, HH]):
17
    ax = fig.add_subplot(1, 4, i + 1)
18
    ax.imshow(a, interpolation="nearest", cmap=plt.cm.gray)
19
    ax.set_title(titles[i], fontsize=10)
20
    ax.set_xticks([])
21
    ax.set_yticks([])
22

23
fig.tight_layout()
24
plt.show()
25