v0lta / PyTorch-Wavelet-ToolboxLinks
Differentiable fast wavelet transforms in PyTorch with GPU support.
☆400Updated last week
Alternatives and similar repositories for PyTorch-Wavelet-Toolbox
Users that are interested in PyTorch-Wavelet-Toolbox are comparing it to the libraries listed below
Sorting:
- Pytorch implementation of 2D Discrete Wavelet (DWT) and Dual Tree Complex Wavelet Transforms (DTCWT) and a DTCWT based ScatterNet☆1,150Updated 2 years ago
- Implementation of 1D, 2D, and 3D FFT convolutions in PyTorch. Much faster than direct convolutions for large kernel sizes.☆515Updated 2 years ago
- Code repository of the paper "Modelling Long Range Dependencies in ND: From Task-Specific to a General Purpose CNN" https://arxiv.org/abs…☆183Updated 7 months ago
- This is an official pytorch implementation of Fast Fourier Convolution.☆382Updated last year
- DCT (discrete cosine transform) functions for pytorch☆639Updated 3 years ago
- Adaptive wavelet pooling for CNN in PyTorch, AISTATS 2021.☆33Updated 4 years ago
- Fast implementation of the multi-dimensional fractional Fourier transform in Python☆50Updated 3 years ago
- Multi-level Wavelet Convolutional Neural Networks☆299Updated 4 years ago
- Implementation of Depthwise Separable Convolution (pytorch)☆83Updated 5 years ago
- PyTorch implementation of the wavelet analysis from Torrence & Compo (1998)☆318Updated 3 years ago
- Adaptive, interpretable wavelets across domains (NeurIPS 2021)☆82Updated 3 years ago
- A python/pytorch package for invertible neural networks☆71Updated 2 years ago
- Convolutional layer for Kolmogorov-Arnold Network (KAN)☆111Updated 9 months ago
- Description: Frequency Augmented Variational Autoencoder for better Image Reconstruction☆44Updated 2 years ago
- Implementation of a U-net complete with efficient attention as well as the latest research findings☆292Updated last year
- It contains the code for the paper Deep Adaptive Wavelet Network☆73Updated 3 years ago
- PyTorch implementations of KMeans, Soft-KMeans and Constrained-KMeans which can be run on GPU and work on (mini-)batches of data.☆76Updated 3 years ago
- A demo of using Hilbert-Huang Transform (HHT) for non-stationary and non-linear signal analysis.☆109Updated 4 years ago
- Official implementation of "DiffuseVAE: Efficient, Controllable and High-Fidelity Generation from Low-Dimensional Latents"☆379Updated 3 years ago
- Official PyTorch implementation for the paper "CARD: Classification and Regression Diffusion Models"☆233Updated 2 years ago
- Fast Convolutional KAN☆66Updated last year
- A simple way to keep track of an Exponential Moving Average (EMA) version of your Pytorch model☆632Updated last week
- DeepInverse: a PyTorch library for solving imaging inverse problems using deep learning☆613Updated this week
- the code for WaveCNet☆134Updated 2 years ago
- 1D interpolation for pytorch☆176Updated last year
- Official Pytorch Implementation of the paper: Wavelet Diffusion Models are fast and scalable Image Generators (CVPR'23)☆434Updated last year
- Wavelet scattering transforms in Python with GPU acceleration☆824Updated 11 months ago
- Code for PnP-Flow: Plug-and-Play image restoration with Flow Matching (ICLR 2025)☆136Updated 3 months ago
- ☆56Updated last week
- A fully invertible U-Net for memory efficiency in Pytorch.☆130Updated 3 years ago