The-AI-Summer / self-attention-cvLinks
Implementation of various self-attention mechanisms focused on computer vision. Ongoing repository. 
☆1,212Updated 4 years ago
Alternatives and similar repositories for self-attention-cv
Users that are interested in self-attention-cv are comparing it to the libraries listed below
Sorting:
- Implementation of the Swin Transformer in PyTorch.☆847Updated 4 years ago
 - Pytorch reimplementation of the Vision Transformer (An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale)☆2,090Updated 3 years ago
 - Recent Transformer-based CV and related works.☆1,334Updated 2 years ago
 - Official implementation of PVT series☆1,862Updated 3 years ago
 - Image Test Time Augmentation with PyTorch!☆1,024Updated 2 years ago
 - ICCV2021, Tokens-to-Token ViT: Training Vision Transformers from Scratch on ImageNet☆1,189Updated 2 years ago
 - PoolFormer: MetaFormer Is Actually What You Need for Vision (CVPR 2022 Oral)☆1,355Updated last year
 - Implementation of ConvMixer for "Patches Are All You Need? 🤷"☆1,077Updated 2 years ago
 - An All-MLP solution for Vision, from Google AI☆1,049Updated 3 months ago
 - This is an official implementation of CvT: Introducing Convolutions to Vision Transformers.☆584Updated 2 years ago
 - Explainability for Vision Transformers☆1,011Updated 3 years ago
 - (ICLR 2022 Spotlight) Official PyTorch implementation of "How Do Vision Transformers Work?"☆819Updated 3 years ago
 - [CVPR 2021] Official PyTorch implementation for Transformer Interpretability Beyond Attention Visualization, a novel method to visualize …☆1,948Updated last year
 - Class activation maps for your PyTorch models (CAM, Grad-CAM, Grad-CAM++, Smooth Grad-CAM++, Score-CAM, SS-CAM, IS-CAM, XGrad-CAM, Layer-…☆2,265Updated last week
 - This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows" on Semantic Segmentation…☆1,240Updated 3 years ago
 - This is an official implementation for "Self-Supervised Learning with Swin Transformers".☆663Updated 4 years ago
 - label-smooth, amsoftmax, partial-fc, focal-loss, triplet-loss, lovasz-softmax. Maybe useful☆2,252Updated last year
 - A PyTorch implementation of "CoAtNet: Marrying Convolution and Attention for All Data Sizes"☆392Updated 4 years ago
 - Collect some papers about transformer with vision. Awesome Transformer with Computer Vision (CV)☆3,545Updated 9 months ago
 - [CVPR 2021] Involution: Inverting the Inherence of Convolution for Visual Recognition, a brand new neural operator☆1,315Updated 4 years ago
 - Collection of common code that's shared among different research projects in FAIR computer vision team.☆2,189Updated 2 months ago
 - Per-Pixel Classification is Not All You Need for Semantic Segmentation (NeurIPS 2021, spotlight)☆1,436Updated 3 years ago
 - Usable Implementation of "Bootstrap Your Own Latent" self-supervised learning, from Deepmind, in Pytorch☆1,852Updated last year
 - Escaping the Big Data Paradigm with Compact Transformers, 2021 (Train your Vision Transformers in 30 mins on CIFAR-10 with a single GPU!)☆536Updated 11 months ago
 - PyTorch implementation of SwAV https//arxiv.org/abs/2006.09882☆2,080Updated 2 years ago
 - Code for the Convolutional Vision Transformer (ConViT)☆470Updated 4 years ago
 - This is a PyTorch re-implementation of Axial-DeepLab (ECCV 2020 Spotlight)☆456Updated 4 years ago
 - PyTorch implementation of MoCo v3 https//arxiv.org/abs/2104.02057☆1,301Updated 3 years ago
 - Official PyTorch code for "BAM: Bottleneck Attention Module (BMVC2018)" and "CBAM: Convolutional Block Attention Module (ECCV2018)"☆2,190Updated 2 years ago
 - PyTorch implementation of SimSiam https//arxiv.org/abs/2011.10566☆1,213Updated 2 years ago