GAMPTeam / vampyre

Approximate Message Passing in Python
MIT License
65 stars 28 forks source link

Added FFT2 transform, fixes to dependencies, fixes to allow wavelet denoising to run #7

Closed hbraunDSP closed 5 years ago

hbraunDSP commented 5 years ago

I've played around a little with vampyre and ended up modifying a few things.

  1. import vampyre doesn't work unless the pywavelets ans scikit-learn packages are installed. I added these to setup.py.

  2. I fixed some problems which prevented me from using vampyre for wavelet denoising. Unfortunately my editor made a lot of changes to the whitespace of the files. The relevant changes are at lines 37-41 and 109 of wavelet.py, and at line 84 of linear.py

  3. I added fft2.py, which implements a 2D fourier transform. This is a work in progress; mostly I still need to add the SVD.

sdrangan commented 5 years ago

Thanks for doing this!

hbraunDSP commented 5 years ago

No problem! As you can see I don't have much time to devote to this but I'd like to see it become a reality.