haar_pytorch: Pytorch implementation of forward and inverse Haar Wavelets 2D
A simple library that implements differentiable forward and inverse Haar Wavelets.
Install package
pip install haar_pytorch
Install the latest version
pip install --upgrade git+https://github.com/bes-dev/haar_pytorch.git
Example
import torch
from haar_pytorch import HaarForward, HaarInverse
haar = HaarForward()
ihaar = HaarInverse()
img = torch.randn(5, 4, 64, 64)
wavelets = haar(img)
img_reconstructed = ihaar(wavelets)