Jongchan / attention-module
Official PyTorch code for "BAM: Bottleneck Attention Module (BMVC2018)" and "CBAM: Convolutional Block Attention Module (ECCV2018)"
☆2,096Updated 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,364Updated last year
- PyTorch implementation of SENet☆2,296Updated 3 years ago
- Code for ECA-Net: Efficient Channel Attention for Deep Convolutional Neural Networks☆1,298Updated 3 years ago
- Dual Attention Network for Scene Segmentation (CVPR2019)☆2,420Updated 3 weeks ago
- (ImageNet pretrained models) The official pytorch implemention of the TPAMI paper "Res2Net: A New Multi-scale Backbone Architecture"☆1,084Updated 2 years ago
- Implementation of Non-local Block.☆1,573Updated last year
- GCNet: Non-local Networks Meet Squeeze-Excitation Networks and Beyond☆1,198Updated 3 years ago
- CCNet: Criss-Cross Attention for Semantic Segmentation (TPAMI 2020 & ICCV 2019).☆1,436Updated 3 years ago
- label-smooth, amsoftmax, partial-fc, focal-loss, triplet-loss, lovasz-softmax. Maybe useful☆2,207Updated 3 months ago
- Code for our CVPR2021 paper coordinate attention☆1,032Updated 3 years ago
- ResNeSt: Split-Attention Networks☆3,248Updated 2 years ago
- a pytorch lib with state-of-the-art architectures, pretrained models and real-time updated results☆862Updated 4 years ago
- Deformable ConvNets V2 (DCNv2) in PyTorch☆1,455Updated 2 years ago
- Official implementation of PVT series☆1,764Updated 2 years ago
- code and trained models for "Attentional Feature Fusion"☆760Updated 3 years ago
- [CVPR 2021] Involution: Inverting the Inherence of Convolution for Visual Recognition, a brand new neural operator☆1,309Updated 3 years ago
- Efficient AI Backbones including GhostNet, TNT and MLP, developed by Huawei Noah's Ark Lab.☆4,115Updated last month
- a pytorch code about Residual Attention Network. This code is based on two projects from☆682Updated 6 years ago
- Implementation of the Swin Transformer in PyTorch.☆814Updated 3 years ago
- PyTorch implementation of Deformable ConvNets v2 (Modulated Deformable Convolution)☆752Updated 5 years ago
- RepVGG: Making VGG-style ConvNets Great Again☆3,360Updated last year
- A CV toolkit for my papers.☆2,047Updated 3 weeks ago
- Train the HRNet model on ImageNet☆977Updated 3 years ago
- Awesome List of Attention Modules and Plug&Play Modules in Computer Vision☆1,132Updated last year
- Implementing Attention Augmented Convolutional Networks using Pytorch☆646Updated 2 years ago
- [CVPR 2021] Rethinking Semantic Segmentation from a Sequence-to-Sequence Perspective with Transformers☆1,056Updated 4 months ago
- ICCV2021, Tokens-to-Token ViT: Training Vision Transformers from Scratch on ImageNet☆1,169Updated last year
- A PyTorch Implementation of Focal Loss.☆971Updated 5 years ago
- Pytorch implementation of Center Loss☆976Updated last year
- pytorch实现Grad-CAM和Grad-CAM++,可以可视化任意分类网络的Class Activation Map (CAM)图,包括自定义的网络;同时也实现了目标检测faster r-cnn和retinanet两个网络的CAM图;欢迎试用、关注并反馈问题...☆740Updated 4 years ago