Official PyTorch code for "BAM: Bottleneck Attention Module (BMVC2018)" and "CBAM: Convolutional Block Attention Module (ECCV2018)"
☆2,209Mar 9, 2023Updated 2 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,483Jul 12, 2023Updated 2 years ago
- Dual Attention Network for Scene Segmentation (CVPR2019)☆2,458Dec 23, 2024Updated last year
- PyTorch implementation of SENet☆2,343Mar 2, 2021Updated 5 years ago
- Code for ECA-Net: Efficient Channel Attention for Deep Convolutional Neural Networks☆1,397May 16, 2021Updated 4 years ago
- GCNet: Non-local Networks Meet Squeeze-Excitation Networks and Beyond☆1,219Feb 16, 2021Updated 5 years ago
- Implementation of Non-local Block.☆1,573Apr 12, 2023Updated 2 years ago
- Unofficial implementation of Convolutional Block Attention Module (CBAM)☆101Sep 6, 2018Updated 7 years ago
- ResNeSt: Split-Attention Networks☆3,267Dec 9, 2022Updated 3 years ago
- Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, I…☆12,643Apr 7, 2025Updated 10 months ago
- Squeeze-and-Excitation Networks☆3,611Feb 25, 2019Updated 7 years ago
- a pytorch code about Residual Attention Network. This code is based on two projects from☆686Oct 29, 2018Updated 7 years ago
- Code for our CVPR2021 paper coordinate attention☆1,126Jun 8, 2021Updated 4 years ago
- Implementing Attention Augmented Convolutional Networks using Pytorch☆656Jan 26, 2022Updated 4 years ago
- a pytorch lib with state-of-the-art architectures, pretrained models and real-time updated results☆871Dec 8, 2020Updated 5 years ago
- CCNet: Criss-Cross Attention for Semantic Segmentation (TPAMI 2020 & ICCV 2019).☆1,481Mar 19, 2021Updated 4 years ago
- A PyTorch implementation of EfficientNet☆8,221Apr 8, 2022Updated 3 years ago
- The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights --…☆36,420Updated this week
- Code for our CVPR 2019 paper: Selective Kernel Networks; See zhihu：https://zhuanlan.zhihu.com/p/59690223☆625Mar 26, 2019Updated 6 years ago
- Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.☆9,112Apr 22, 2022Updated 3 years ago
- 🍀 Pytorch implementation of various Attention Mechanisms, MLP, Re-parameter, Convolution, which is helpful to further understand papers.…☆12,162Dec 6, 2024Updated last year
- Pytorch implementation of convolutional neural network visualization techniques☆8,200Jan 1, 2025Updated last year
- ☆231Mar 1, 2019Updated 7 years ago
- Please choose the openseg.pytorch project for the updated code that achieve SOTA on 6 benchmarks!☆812Feb 16, 2021Updated 5 years ago
- A CV toolkit for my papers.☆2,049Dec 21, 2024Updated last year
- cvpr2024/cvpr2023/cvpr2022/cvpr2021/cvpr2020/cvpr2019/cvpr2018/cvpr2017 论文/代码/解读/直播合集，极市团队整理☆12,507Apr 25, 2024Updated last year
- OpenMMLab Detection Toolbox and Benchmark☆32,418Aug 21, 2024Updated last year
- Pytorch implementation of Self-Attention Generative Adversarial Networks (SAGAN)☆2,611Apr 22, 2024Updated last year
- Efficient AI Backbones including GhostNet, TNT and MLP, developed by Huawei Noah's Ark Lab.☆4,385Mar 15, 2025Updated 11 months ago
- Count the MACs / FLOPs of your PyTorch model.☆5,082Jul 8, 2024Updated last year
- Use of Attention Gates in a Convolutional Neural Network / Medical Image Classification and Segmentation☆2,040Nov 13, 2020Updated 5 years ago
- This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".☆15,721Jul 24, 2024Updated last year
- Deformable ConvNets V2 (DCNv2) in PyTorch☆1,491Nov 18, 2022Updated 3 years ago
- Deformable Convolutional Networks☆4,116Sep 27, 2021Updated 4 years ago
- Implementation of Pyramid Attention Networks for Semantic Segmentation.☆238Dec 19, 2018Updated 7 years ago
- label-smooth, amsoftmax, partial-fc, focal-loss, triplet-loss, lovasz-softmax. Maybe useful☆2,260Oct 17, 2024Updated last year
- FCOS: Fully Convolutional One-Stage Object Detection (ICCV'19)☆3,347Dec 9, 2023Updated 2 years ago
- Object detection, 3D detection, and pose estimation using center point detection:☆7,546Mar 2, 2023Updated 3 years ago
- Non-local Neural Networks for Video Classification☆1,996Sep 15, 2021Updated 4 years ago
- DeepLab v3+ model in PyTorch. Support different backbones.☆3,004Aug 4, 2024Updated last year