Jongchan / attention-module
Official PyTorch code for "BAM: Bottleneck Attention Module (BMVC2018)" and "CBAM: Convolutional Block Attention Module (ECCV2018)"
☆2,062Updated last year
Related projects ⓘ
Alternatives and complementary repositories for attention-module
- Non-official implement of Paper：CBAM: Convolutional Block Attention Module☆1,337Updated last year
- PyTorch implementation of SENet☆2,285Updated 3 years ago
- Code for ECA-Net: Efficient Channel Attention for Deep Convolutional Neural Networks☆1,269Updated 3 years ago
- Dual Attention Network for Scene Segmentation (CVPR2019)☆2,408Updated last year
- (ImageNet pretrained models) The official pytorch implemention of the TPAMI paper "Res2Net: A New Multi-scale Backbone Architecture"☆1,074Updated last year
- Implementation of Non-local Block.☆1,570Updated last year
- GCNet: Non-local Networks Meet Squeeze-Excitation Networks and Beyond☆1,193Updated 3 years ago
- Code for our CVPR2021 paper coordinate attention☆1,017Updated 3 years ago
- CCNet: Criss-Cross Attention for Semantic Segmentation (TPAMI 2020 & ICCV 2019).☆1,426Updated 3 years ago
- ResNeSt: Split-Attention Networks☆3,238Updated last year
- Deformable ConvNets V2 (DCNv2) in PyTorch☆1,443Updated 2 years ago
- label-smooth, amsoftmax, partial-fc, focal-loss, triplet-loss, lovasz-softmax. Maybe useful☆2,183Updated last month
- PyTorch implementation of Deformable ConvNets v2 (Modulated Deformable Convolution)☆742Updated 5 years ago
- [CVPR 2021] Involution: Inverting the Inherence of Convolution for Visual Recognition, a brand new neural operator☆1,309Updated 3 years ago
- ACNet: Strengthening the Kernel Skeletons for Powerful CNN via Asymmetric Convolution Blocks☆831Updated last year
- code and trained models for "Attentional Feature Fusion"☆742Updated 3 years ago
- a pytorch lib with state-of-the-art architectures, pretrained models and real-time updated results☆860Updated 3 years ago
- The OCR approach is rephrased as Segmentation Transformer: https://arxiv.org/abs/1909.11065. This is an official implementation of semant…☆3,160Updated 2 years ago
- Official implementation of PVT series☆1,729Updated 2 years ago
- Awesome List of Attention Modules and Plug&Play Modules in Computer Vision☆1,099Updated last year
- A CV toolkit for my papers.☆2,041Updated 5 months ago
- Flops counter for convolutional networks in pytorch framework☆2,822Updated last month
- a pytorch code about Residual Attention Network. This code is based on two projects from☆680Updated 6 years ago
- Train the HRNet model on ImageNet☆970Updated 2 years ago
- PyTorch Implementation of Fully Convolutional Networks. (Training code to reproduce the original result is available.)☆1,748Updated last year
- pytorch实现Grad-CAM和Grad-CAM++,可以可视化任意分类网络的Class Activation Map (CAM)图,包括自定义的网络;同时也实现了目标检测faster r-cnn和retinanet两个网络的CAM图;欢迎试用、关注并反馈问题...☆728Updated 3 years ago
- Implementation of the Swin Transformer in PyTorch.☆794Updated 3 years ago
- RepVGG: Making VGG-style ConvNets Great Again☆3,333Updated last year
- Efficient AI Backbones including GhostNet, TNT and MLP, developed by Huawei Noah's Ark Lab.☆4,056Updated 4 months ago
- [CVPR 2021] Rethinking Semantic Segmentation from a Sequence-to-Sequence Perspective with Transformers☆1,054Updated 2 months ago