Jongchan / attention-module
Official PyTorch code for "BAM: Bottleneck Attention Module (BMVC2018)" and "CBAM: Convolutional Block Attention Module (ECCV2018)"
☆2,119Updated 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
- Non-official implement of Paper:CBAM: Convolutional Block Attention Module☆1,376Updated last year
- PyTorch implementation of SENet☆2,315Updated 4 years ago
- Code for ECA-Net: Efficient Channel Attention for Deep Convolutional Neural Networks☆1,316Updated 3 years ago
- Dual Attention Network for Scene Segmentation (CVPR2019)☆2,425Updated 3 months ago
- CCNet: Criss-Cross Attention for Semantic Segmentation (TPAMI 2020 & ICCV 2019).☆1,452Updated 4 years ago
- Implementation of Non-local Block.☆1,571Updated last year
- (ImageNet pretrained models) The official pytorch implemention of the TPAMI paper "Res2Net: A New Multi-scale Backbone Architecture"☆1,090Updated 2 years ago
- GCNet: Non-local Networks Meet Squeeze-Excitation Networks and Beyond☆1,203Updated 4 years ago
- Code for our CVPR2021 paper coordinate attention☆1,042Updated 3 years ago
- ResNeSt: Split-Attention Networks☆3,246Updated 2 years ago
- Deformable ConvNets V2 (DCNv2) in PyTorch☆1,459Updated 2 years ago
- [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)☆754Updated 5 years ago
- Implementation of the Swin Transformer in PyTorch.☆821Updated 3 years ago
- label-smooth, amsoftmax, partial-fc, focal-loss, triplet-loss, lovasz-softmax. Maybe useful☆2,223Updated 5 months ago
- Official implementation of PVT series☆1,787Updated 2 years ago
- RepVGG: Making VGG-style ConvNets Great Again☆3,377Updated 2 years ago
- pytorch实现Grad-CAM和Grad-CAM++,可以可视化任意分类网络的Class Activation Map (CAM)图,包括自定义的网络;同时也实现了目标检测faster r-cnn和retinanet两个网络的CAM图;欢迎试用、关注并反馈问题...☆744Updated 4 years ago
- Train the HRNet model on ImageNet☆992Updated 3 years ago
- code and trained models for "Attentional Feature Fusion"☆766Updated 3 years ago
- Efficient AI Backbones including GhostNet, TNT and MLP, developed by Huawei Noah's Ark Lab.☆4,165Updated last week
- a pytorch lib with state-of-the-art architectures, pretrained models and real-time updated results☆863Updated 4 years ago
- a pytorch code about Residual Attention Network. This code is based on two projects from☆682Updated 6 years ago
- Implementing Attention Augmented Convolutional Networks using Pytorch☆649Updated 3 years ago
- Flops counter for convolutional networks in pytorch framework☆2,869Updated 2 months ago
- A PyTorch Implementation of Focal Loss.☆979Updated 5 years ago
- Squeeze-and-Excitation Networks☆3,470Updated 6 years ago
- DeepLab v3+ model in PyTorch. Support different backbones.☆2,946Updated 7 months ago
- ACNet: Strengthening the Kernel Skeletons for Powerful CNN via Asymmetric Convolution Blocks☆840Updated 2 years ago
- A memory-efficient implementation of DenseNets☆1,529Updated last year