search

fbcotter / pytorch_wavelets

Pytorch implementation of 2D Discrete Wavelet (DWT) and Dual Tree Complex Wavelet Transforms (DTCWT) and a DTCWT based ScatterNet
Other
985 stars 148 forks source link

3D wavelet transform #56

Open RY-97 opened 6 months ago

RY-97 commented 6 months ago

I was doing a similar job recently and encountered the same problem. I would like to ask for your help. I want to input a (b, c, w, h, d) 3D image data. The output after discrete wavelet transform is still consistent in shape, but does not require inverse wavelet transform. Do you have any solution?

vectorzwt commented 6 months ago

You can check this repo: https://github.com/KeKsBoTer/torch-dwt.