bes-dev/haar_pytorch

Pytorch implementation of forward and inverse Haar Wavelets 2D

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)