Getting started

First you can import some modules like:

import pywt
import numpy as np

import statsWaveletFilt.filtration as fil
import statsWaveletFilt.signals as sig
import statsWaveletFilt.miscellaneous as misc

And follow some steps to make your filtrations:

First step: Signal modelling

In many test for filtration and/or estimation of error, showed in articles, are supose an known ideal signal, how show the code above:

import matplotlib.pyplot as plt

dimension = 1024

t, idealSignal = sig.dopplerFunction(dimension)
plt.plot(t, idealSignal, 'k-')
plt.show()
img1

This package also gives a support for anothers three ideal signal: the heavsine (sig.heavsineFunction()), bump (sig.bumpFunction()) and block (sig.blockFunction()).

And add to this a kind of noise, generally gaussian or, also called, normal.

img2

Second step: Wavelet transformation

Let’s do this using the pywt.wavedec function. With this you can 1) make a wavelet transform of your noisy signal, 2) choice wich wavelet function is used ('haar', 'db', 'gaus', 'mexh' among others) and 3) choice any levels you want to evaluate the transform.

The first position = noiseCoeff[0] has the scale coefficents of the last nível. The consecutive elements `` = noiseCoeff[1:]`` has the wavelet coefficients of the last to the first level of transform.

Using misc.showWaveletCoeff function you can show all coefficients returned of the transform.

img3

Thirth step: Filtration

With the fil.filtration function you can put yours 1) coefficients for filtration, 2) choice the method ('visu', 'sure', 'bayes', 'spc') and, if necessary, 3) set method own parameters (more details in documentation).

And, again, you can show the wavelet coefficients before and after the filtration, with the limiar used using the misc.showWaveletCoeff function.

img4
img5

Fourth step: Recovery wavelet transformation Using the pywt.waverec function we can revovery the signal.

img6

Fifth step: Figures of merith Using sig.differential_snr_dB we can avaliate the filtration using some figures of merith of different forms of evaluate the snr or cnr (see more in Documentation) of the signal.

sig.differential_snr_dB(noisySignal, filtratedSignal, method=’variances’, idealSignal=idealSignal)

Appendix: Miscelaneous

  • The misc.generateData function generate data using the signals showed here with different variances of noise and save in format .npy.
  • The misc.normalizeData function normalize a data between a min and max value, this function is used in all signals modeled in sig module.