search

fbcotter / pytorch_wavelets

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

Is the SWT avaliable now? Is there something wrong in the SWTInverse under the current version? #15

Closed yeelinsen closed 4 years ago

yeelinsen commented 4 years ago

Hi, @fbcotter .When I used the SWTForward in transpose2d.py and SWTInverse in swt_inverse.py, I found that the output of SWTInverse with input(N, 4C, H, W) has a shape of NC2H*2W. Shouldn't it be the shape (N, C, H, W) ? Thank u for this helpful repository and the great effort you've paid!

fbcotter commented 4 years ago

Hi @yeelinsen, the SWT implementation in my repo is very untested. I'd like to properly sit down and make it work nicely. I think the Inverse SWT has many possible options too, as it's a redundant transform, and I haven't yet chosen which to implement

yeelinsen commented 4 years ago

Thank you for your patient apply!