Official PyTorch code for "BAM: Bottleneck Attention Module (BMVC2018)" and "CBAM: Convolutional Block Attention Module (ECCV2018)"
☆2,211Mar 9, 2023Updated 3 years ago
Alternatives and similar repositories for attention-module
Users that are interested in attention-module are comparing it to the libraries listed below
Sorting:
- Non-official implement of Paper：CBAM: Convolutional Block Attention Module☆1,486Jul 12, 2023Updated 2 years ago
- PyTorch implementation of SENet☆2,345Mar 2, 2021Updated 5 years ago
- Dual Attention Network for Scene Segmentation (CVPR2019)☆2,457Dec 23, 2024Updated last year
- Code for ECA-Net: Efficient Channel Attention for Deep Convolutional Neural Networks☆1,398May 16, 2021Updated 4 years ago
- GCNet: Non-local Networks Meet Squeeze-Excitation Networks and Beyond☆1,221Feb 16, 2021Updated 5 years ago
- Unofficial implementation of Convolutional Block Attention Module (CBAM)☆101Sep 6, 2018Updated 7 years ago
- Implementation of Non-local Block.☆1,568Apr 12, 2023Updated 2 years ago
- Squeeze-and-Excitation Networks☆3,614Feb 25, 2019Updated 7 years ago
- Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, I…☆12,701Apr 7, 2025Updated 11 months ago
- ResNeSt: Split-Attention Networks☆3,264Dec 9, 2022Updated 3 years ago
- Code for our CVPR2021 paper coordinate attention☆1,132Jun 8, 2021Updated 4 years ago
- a pytorch code about Residual Attention Network. This code is based on two projects from☆686Oct 29, 2018Updated 7 years ago
- Implementing Attention Augmented Convolutional Networks using Pytorch☆654Jan 26, 2022Updated 4 years ago
- CCNet: Criss-Cross Attention for Semantic Segmentation (TPAMI 2020 & ICCV 2019).☆1,484Mar 19, 2021Updated 5 years ago
- 🍀 Pytorch implementation of various Attention Mechanisms, MLP, Re-parameter, Convolution, which is helpful to further understand papers.…☆12,170Updated this week
- The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights --…☆36,504Mar 13, 2026Updated last week
- ☆229Mar 1, 2019Updated 7 years ago
- A PyTorch implementation of EfficientNet☆8,219Apr 8, 2022Updated 3 years ago
- Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.☆9,112Apr 22, 2022Updated 3 years ago
- Pytorch implementation of Self-Attention Generative Adversarial Networks (SAGAN)☆2,609Apr 22, 2024Updated last year
- cvpr2024/cvpr2023/cvpr2022/cvpr2021/cvpr2020/cvpr2019/cvpr2018/cvpr2017 论文/代码/解读/直播合集，极市团队整理☆12,506Apr 25, 2024Updated last year
- Use of Attention Gates in a Convolutional Neural Network / Medical Image Classification and Segmentation☆2,045Nov 13, 2020Updated 5 years ago
- Please choose the openseg.pytorch project for the updated code that achieve SOTA on 6 benchmarks!☆812Feb 16, 2021Updated 5 years ago
- Implementation of Pyramid Attention Networks for Semantic Segmentation.☆238Dec 19, 2018Updated 7 years ago
- Count the MACs / FLOPs of your PyTorch model.☆5,080Jul 8, 2024Updated last year
- Pytorch implementation of convolutional neural network visualization techniques☆8,199Jan 1, 2025Updated last year
- This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".☆15,767Jul 24, 2024Updated last year
- Efficient AI Backbones including GhostNet, TNT and MLP, developed by Huawei Noah's Ark Lab.☆4,393Mar 15, 2025Updated last year
- A CV toolkit for my papers.☆2,048Dec 21, 2024Updated last year
- CBAM implementation on TensowFlow☆150Aug 31, 2018Updated 7 years ago
- Code for our ICASSP 2021 paper: SA-Net: Shuffle Attention for Deep Convolutional Neural Networks☆266May 29, 2021Updated 4 years ago
- try to implement the CVPR 2019 paper "Selective Kernel Networks" by PyTorch☆381Oct 21, 2019Updated 6 years ago
- Deformable ConvNets V2 (DCNv2) in PyTorch☆1,489Nov 18, 2022Updated 3 years ago
- label-smooth, amsoftmax, partial-fc, focal-loss, triplet-loss, lovasz-softmax. Maybe useful☆2,262Oct 17, 2024Updated last year
- CBAM implementation on Keras☆368Feb 8, 2021Updated 5 years ago
- Deformable Convolutional Networks☆4,116Sep 27, 2021Updated 4 years ago
- Non-local Neural Networks for Video Classification☆1,994Sep 15, 2021Updated 4 years ago
- DeepLab v3+ model in PyTorch. Support different backbones.☆3,006Aug 4, 2024Updated last year
- Object detection, 3D detection, and pose estimation using center point detection:☆7,546Mar 2, 2023Updated 3 years ago