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
259 stars 34 forks source link

Support 3D-Stationary Wavelet Transform? #86

Open vectorzwt opened 1 month ago

vectorzwt commented 1 month ago

Hi,

thank you for providing this toolbox. Is there any future work planned toward allowing 3D stationary wavelet transform? E.g. on an array of shape (batch, channels, height, width, depth) .

v0lta commented 1 month ago

Work is currently focused on the 1D and 2D cases. When that's done, we can look at 3D, which will take time.