search

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

Support for 3D-Wavelet decompositions? #30

Closed homerjed closed 2 years ago

homerjed commented 2 years ago

Hi - this looks really cool! Just wondering if there is any plan for wavelet decompositions for 3D tensors?

v0lta commented 2 years ago

For the moment I think the priority must be to extend test coverage to further improve the code we already have. In the long run, absolutely yes! The goal of this project is to provide Pytorch code that covers PyWavelet functionality completely. This is certainly something I would love to see in the 1.0 release.

v0lta commented 2 years ago

I have started to implement 3d transformations in the https://github.com/v0lta/PyTorch-Wavelet-Toolbox/tree/3dwavedec branch.

v0lta commented 2 years ago

https://github.com/v0lta/PyTorch-Wavelet-Toolbox/releases/tag/v0.1.2 now ships this functionality.

homerjed commented 2 years ago

Incredible! Thank you. I will try it out now!

v0lta commented 2 years ago

Thanks, if you run into problems or observe any unexpected behaviour please let me know.