Jongchan / attention-moduleLinks
Official PyTorch code for "BAM: Bottleneck Attention Module (BMVC2018)" and "CBAM: Convolutional Block Attention Module (ECCV2018)"
☆2,162Updated 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,421Updated last year
- PyTorch implementation of SENet☆2,324Updated 4 years ago
- Code for ECA-Net: Efficient Channel Attention for Deep Convolutional Neural Networks☆1,340Updated 4 years ago
- Dual Attention Network for Scene Segmentation (CVPR2019)☆2,436Updated 6 months ago
- (ImageNet pretrained models) The official pytorch implemention of the TPAMI paper "Res2Net: A New Multi-scale Backbone Architecture"☆1,101Updated 2 years ago
- Implementation of Non-local Block.☆1,574Updated 2 years ago
- CCNet: Criss-Cross Attention for Semantic Segmentation (TPAMI 2020 & ICCV 2019).☆1,466Updated 4 years ago
- GCNet: Non-local Networks Meet Squeeze-Excitation Networks and Beyond☆1,209Updated 4 years ago
- ResNeSt: Split-Attention Networks☆3,257Updated 2 years ago
- Deformable ConvNets V2 (DCNv2) in PyTorch☆1,473Updated 2 years ago
- Code for our CVPR2021 paper coordinate attention☆1,068Updated 4 years ago
- Train the HRNet model on ImageNet☆1,019Updated 3 years ago
- a pytorch code about Residual Attention Network. This code is based on two projects from☆684Updated 6 years ago
- code and trained models for "Attentional Feature Fusion"☆789Updated 3 years ago
- [CVPR 2021] Involution: Inverting the Inherence of Convolution for Visual Recognition, a brand new neural operator☆1,312Updated 3 years ago
- a pytorch lib with state-of-the-art architectures, pretrained models and real-time updated results☆866Updated 4 years ago
- Implementing Attention Augmented Convolutional Networks using Pytorch☆654Updated 3 years ago
- Implementation of the Swin Transformer in PyTorch.☆836Updated 4 years ago
- label-smooth, amsoftmax, partial-fc, focal-loss, triplet-loss, lovasz-softmax. Maybe useful☆2,245Updated 8 months ago
- PyTorch implementation of Deformable ConvNets v2 (Modulated Deformable Convolution)☆756Updated 6 years ago
- Official implementation of PVT series☆1,826Updated 2 years ago
- Code for our CVPR 2019 paper: Selective Kernel Networks; See zhihu：https://zhuanlan.zhihu.com/p/59690223☆610Updated 6 years ago
- pytorch实现Grad-CAM和Grad-CAM++,可以可视化任意分类网络的Class Activation Map (CAM)图,包括自定义的网络;同时也实现了目标检测faster r-cnn和retinanet两个网络的CAM图;欢迎试用、关注并反馈问题...☆754Updated 4 years ago
- Awesome List of Attention Modules and Plug&Play Modules in Computer Vision☆1,206Updated 2 years ago
- RepVGG: Making VGG-style ConvNets Great Again☆3,402Updated 2 years ago
- ACNet: Strengthening the Kernel Skeletons for Powerful CNN via Asymmetric Convolution Blocks☆845Updated 2 years ago
- Exploring Self-attention for Image Recognition, CVPR2020.☆752Updated 5 years ago
- PyTorch implementation of image classification models for CIFAR-10/CIFAR-100/MNIST/FashionMNIST/Kuzushiji-MNIST/ImageNet☆1,409Updated 3 years ago
- Flops counter for neural networks in pytorch framework☆2,909Updated 5 months ago
- Squeeze-and-Excitation Networks☆3,538Updated 6 years ago