Jongchan / attention-module
Official PyTorch code for "BAM: Bottleneck Attention Module (BMVC2018)" and "CBAM: Convolutional Block Attention Module (ECCV2018)"
☆2,103Updated last year
Alternatives and similar repositories for attention-module:
Users that are interested in attention-module are comparing it to the libraries listed below
- Non-official implement of Paper：CBAM: Convolutional Block Attention Module☆1,369Updated last year
- PyTorch implementation of SENet☆2,304Updated 3 years ago
- Code for ECA-Net: Efficient Channel Attention for Deep Convolutional Neural Networks☆1,304Updated 3 years ago
- Implementation of Non-local Block.☆1,572Updated last year
- (ImageNet pretrained models) The official pytorch implemention of the TPAMI paper "Res2Net: A New Multi-scale Backbone Architecture"☆1,085Updated 2 years ago
- Dual Attention Network for Scene Segmentation (CVPR2019)☆2,420Updated last month
- ResNeSt: Split-Attention Networks☆3,249Updated 2 years ago
- GCNet: Non-local Networks Meet Squeeze-Excitation Networks and Beyond☆1,201Updated 4 years ago
- Deformable ConvNets V2 (DCNv2) in PyTorch☆1,456Updated 2 years ago
- CCNet: Criss-Cross Attention for Semantic Segmentation (TPAMI 2020 & ICCV 2019).☆1,440Updated 3 years ago
- label-smooth, amsoftmax, partial-fc, focal-loss, triplet-loss, lovasz-softmax. Maybe useful☆2,211Updated 4 months ago
- Code for our CVPR2021 paper coordinate attention☆1,038Updated 3 years ago
- Train the HRNet model on ImageNet☆984Updated 3 years ago
- Official implementation of PVT series☆1,773Updated 2 years ago
- Implementation of the Swin Transformer in PyTorch.☆816Updated 3 years ago
- code and trained models for "Attentional Feature Fusion"☆763Updated 3 years ago
- mobilenetv3 with pytorch，provide pre-train model☆1,683Updated last year
- [CVPR 2021] Involution: Inverting the Inherence of Convolution for Visual Recognition, a brand new neural operator☆1,309Updated 3 years ago
- PyTorch implementation of Deformable ConvNets v2 (Modulated Deformable Convolution)☆752Updated 5 years ago
- Classification with PyTorch.☆1,710Updated 8 months ago
- a pytorch lib with state-of-the-art architectures, pretrained models and real-time updated results☆863Updated 4 years ago
- Implementing Attention Augmented Convolutional Networks using Pytorch☆649Updated 3 years ago
- pytorch实现Grad-CAM和Grad-CAM++,可以可视化任意分类网络的Class Activation Map (CAM)图,包括自定义的网络;同时也实现了目标检测faster r-cnn和retinanet两个网络的CAM图;欢迎试用、关注并反馈问题...☆741Updated 4 years ago
- a pytorch code about Residual Attention Network. This code is based on two projects from☆682Updated 6 years ago
- Efficient AI Backbones including GhostNet, TNT and MLP, developed by Huawei Noah's Ark Lab.☆4,134Updated 2 months ago
- Pytorch reimplementation of the Vision Transformer (An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale)☆1,990Updated 2 years ago
- Flops counter for convolutional networks in pytorch framework☆2,857Updated last month
- Pytorch implementation of Feature Pyramid Network (FPN) for Object Detection☆956Updated 6 years ago
- RepVGG: Making VGG-style ConvNets Great Again☆3,374Updated 2 years ago
- Squeeze-and-Excitation Networks☆3,448Updated 5 years ago