import matplotlib.pyplot as plt
from skued import diffread, baseline_dwt

im = diffread('data/hotspots.tif')
denoised = baseline_dwt(im, max_iter = 250, level = 1, wavelet = 'sym2', axis = (0, 1))

fig, (ax1, ax2) = plt.subplots(1, 2)
ax1.imshow(im, vmin = 0, vmax = 2e3, cmap='inferno')
ax2.imshow(denoised, vmin = 0, vmax = 2e3, cmap='inferno')

for ax in (ax1, ax2):
        ax.xaxis.set_visible(False)
        ax.yaxis.set_visible(False)
plt.tight_layout()
plt.show()