lucidrains / axial-attentionLinks
Implementation of Axial attention - attending to multi-dimensional data efficiently
☆381Updated 3 years ago
Alternatives and similar repositories for axial-attention
Users that are interested in axial-attention are comparing it to the libraries listed below
Sorting:
- This is a PyTorch re-implementation of Axial-DeepLab (ECCV 2020 Spotlight)☆455Updated 3 years ago
- An implementation of the efficient attention module.☆315Updated 4 years ago
- Self-supervised vIsion Transformer (SiT)☆332Updated 2 years ago
- An All-MLP solution for Vision, from Google AI☆1,024Updated 8 months ago
- Unofficial implementation of MLP-Mixer: An all-MLP Architecture for Vision☆217Updated 4 years ago
- [NeurIPS 2021] [T-PAMI] Global Filter Networks for Image Classification☆477Updated last year
- Implementation of a U-net complete with efficient attention as well as the latest research findings☆282Updated last year
- Implementation of Pixel-level Contrastive Learning, proposed in the paper "Propagate Yourself", in Pytorch☆259Updated 4 years ago
- Implementation of Linformer for Pytorch☆286Updated last year
- Implementation of Transformer in Transformer, pixel level attention paired with patch level attention for image classification, in Pytorc…☆305Updated 3 years ago
- A PyTorch implementation of the 1d and 2d Sinusoidal positional encoding/embedding.☆251Updated 4 years ago
- Neighborhood Attention Transformer, arxiv 2022 / CVPR 2023. Dilated Neighborhood Attention Transformer, arxiv 2022☆1,118Updated last year
- This is an official implementation for "Self-Supervised Learning with Swin Transformers".☆656Updated 4 years ago
- ☆246Updated 3 years ago
- Implementation of 1D, 2D, and 3D FFT convolutions in PyTorch. Much faster than direct convolutions for large kernel sizes.☆497Updated last year
- PyTorch Implementation of CvT: Introducing Convolutions to Vision Transformers☆225Updated 4 years ago
- [ICLR'22 Oral] Implementation of "CycleMLP: A MLP-like Architecture for Dense Prediction"☆288Updated 3 years ago
- Label-Efficient Semantic Segmentation with Diffusion Models (ICLR'2022)☆699Updated 2 years ago
- Learning Rate Warmup in PyTorch☆410Updated 2 months ago
- Two simple and effective designs of vision transformer, which is on par with the Swin transformer☆601Updated 2 years ago
- Implementation of ConvMixer for "Patches Are All You Need? 🤷"☆1,072Updated 2 years ago
- Tiny PyTorch library for maintaining a moving average of a collection of parameters.☆429Updated 8 months ago
- Implementation of Segformer, Attention + MLP neural network for segmentation, in Pytorch☆367Updated 2 years ago
- Code for the Convolutional Vision Transformer (ConViT)☆465Updated 3 years ago
- [NeurIPS 2021 Spotlight] Official code for "Focal Self-attention for Local-Global Interactions in Vision Transformers"☆556Updated 3 years ago
- CSWin Transformer: A General Vision Transformer Backbone with Cross-Shaped, CVPR 2022☆569Updated last year
- Implementation of the Swin Transformer in PyTorch.☆830Updated 4 years ago
- A PyTorch implementation of "CoAtNet: Marrying Convolution and Attention for All Data Sizes"☆387Updated 3 years ago
- This is an official implementation of CvT: Introducing Convolutions to Vision Transformers.☆577Updated 2 years ago
- ☆459Updated 2 years ago