UiO-CS / tf-wavelets

TensorFlow implementation of descrete wavelets transforms
MIT License
49 stars 11 forks source link

Would there be a way to use batches? #3

Open megalinier opened 5 years ago

megalinier commented 5 years ago

Hello, and thanks again for your code!

I would like to use your tfw functions to compute the wavelet transform of a whole batch of images in one shot. I have tried but I don't see a way to do so given that your dwt2d relies on dwt1d, which already uses the rows of the 2d image as batches (if I understood correctly). Would you have a suggestion to help me?

Thank you!