f-dangel / unfoldNd
(N=1,2,3)-dimensional unfold (im2col) and fold (col2im) in PyTorch
☆95Updated 3 months ago
Alternatives and similar repositories for unfoldNd:
Users that are interested in unfoldNd are comparing it to the libraries listed below
- Implementation of Uformer, Attention-based Unet, in Pytorch☆93Updated 3 years ago
- [ICLR 2022] "Anti-Oversmoothing in Deep Vision Transformers via the Fourier Domain Analysis: From Theory to Practice" by Peihao Wang, Wen…☆80Updated last year
- A fully invertible U-Net for memory efficiency in Pytorch.☆124Updated 2 years ago
- ☆119Updated 3 years ago
- Official code for paper "On the Connection between Local Attention and Dynamic Depth-wise Convolution" ICLR 2022 Spotlight☆184Updated 2 years ago
- Framework for creating (partially) reversible neural networks with PyTorch☆151Updated 2 years ago
- Implementation of a U-net complete with efficient attention as well as the latest research findings☆276Updated 10 months ago
- Implementation of 1D, 2D, and 3D FFT convolutions in PyTorch. Much faster than direct convolutions for large kernel sizes.☆493Updated last year
- Code repository for the ICLR 2022 paper "FlexConv: Continuous Kernel Convolutions With Differentiable Kernel Sizes" https://openreview.ne…☆115Updated 2 years ago
- Normalized Cross-Correlation in pytorch☆42Updated 2 years ago
- A PyTorch implementation of Connected Components Labeling☆112Updated last year
- Unofficial Implementation of MLP-Mixer, gMLP, resMLP, Vision Permutator, S2MLP, S2MLPv2, RaftMLP, HireMLP, ConvMLP, AS-MLP, SparseMLP, Co…☆169Updated 2 years ago
- ☆190Updated 2 years ago
- [CVPR2022 - Oral] Official Jax Implementation of Learned Queries for Efficient Local Attention☆116Updated 2 years ago
- 3D VQ-VAE-2 for high-resolution CT scan synthesis☆46Updated 3 years ago
- Unofficial implementation of MLP-Mixer: An all-MLP Architecture for Vision☆217Updated 3 years ago
- Official code for ICCV 2023 paper "Convolutional Networks with Oriented 1D Kernels"☆47Updated last year
- [ICLR'22] This is an official implementation for "AS-MLP: An Axial Shifted MLP Architecture for Vision".☆126Updated 2 years ago
- PyTorch reimplementation of FlexiViT: One Model for All Patch Sizes☆60Updated 10 months ago
- Similarity Measure. Re-implementation for PyTorch.☆51Updated 4 years ago
- High-order spline interpolation in PyTorch☆67Updated 6 months ago
- Implementation of bi-directional Conv LSTM and Conv GRU in PyTorch.☆39Updated 6 years ago
- ☆60Updated 3 years ago
- A PyTorch add-on for working with image mappings and displacement fields, including Spatial Transformers☆52Updated 3 years ago
- Optical flow library, based on PyTorch tensors☆20Updated 2 years ago
- A library for calculating the FLOPs in the forward() process based on torch.fx☆102Updated this week
- A better PyTorch implementation of image local attention which reduces the GPU memory by an order of magnitude.☆139Updated 3 years ago
- Official implementation for "SimA: Simple Softmax-free Attention for Vision Transformers"☆43Updated 11 months ago
- Rudimentary Conv4D Layer Implementation for PyTorch☆38Updated 6 years ago
- [ECCV 2022] EdgeViT: Competing Light-weight CNNs on Mobile Devices with Vision Transformers☆107Updated 2 years ago