Improve packets order interface

v0lta / PyTorch-Wavelet-Toolbox

Differentiable fast wavelet transforms in PyTorch with GPU support.

https://pytorch-wavelet-toolbox.readthedocs.io

European Union Public License 1.2

279 stars 36 forks source link

Improve packets order interface #95

Closed felixblanke closed 3 months ago

felixblanke commented 3 months ago

Makes WaveletPacket.get_level static
Adds Option for WaveletPacket.get_level to return the 1d natural order.
Adds WaveletPacket2D.get_level to have a consistent interface (with our 1d case as well as pywt)
Adds unit tests for both 1d orders and 2d natural order

v0lta commented 3 months ago

Thanks, I like this PR. It introduces pywt compatible interfaces, which will make it easier to switch. I merged the main, and so far, all tests have checked out.