attention-module
Attention module library
Provides implementations of attention modules for computer vision tasks using PyTorch
Official PyTorch code for "BAM: Bottleneck Attention Module (BMVC2018)" and "CBAM: Convolutional Block Attention Module (ECCV2018)"
2k stars
19 watching
402 forks
Language: Python
last commit: over 1 year ago
Linked from 1 awesome list
weiaicunzai/awesome-image-classificationRelated projects:
| Repository | Description | Stars |
|---|---|---|
 huyz1117/bam | An implementation of the Bottleneck Attention Module in TensorFlow using attention mechanism | 12 |
 elbuco1/cbam | A deep learning project that develops and tests a novel attention mechanism called CBAM for image classification tasks. | 104 |
 luuuyi/cbam.pytorch | PyTorch implementation of the CBAM module for refining feature maps in deep networks | 1,337 |
 kaiyangzhou/dassl.pytorch | A PyTorch toolbox for supporting research and development of domain adaptation, generalization, and semi-supervised learning methods in computer vision. | 1,217 |
 kobiso/cbam-keras | A repository implementing a Keras-based architecture for improving neural network representation power with attention mechanisms and squeeze-and-excitation networks. | 361 |
 tengshaofeng/residualattentionnetwork-pytorch | Implementing a deep learning framework for image classification using Residual Attention Network architecture | 680 |
 javeywang/pyramid-attention-networks-pytorch | An implementation of a deep learning model using PyTorch for semantic segmentation tasks. | 235 |
 woozzu/dong_iccv_2017 | An implementation of semantic image synthesis via adversarial learning using PyTorch | 145 |
 zhanghang1989/pytorch-encoding | A Python framework for building deep learning models with optimized encoding layers and batch normalization. | 2,041 |
 ngxbac/gain | A PyTorch implementation of an attention-guided inference network to focus on specific areas of objects in images | 48 |
 pp00704831/banet-tip-2022 | A PyTorch implementation of an attention network for dynamic scene deblurring | 37 |
 byungkwanlee/moai | Improves performance of vision language tasks by integrating computer vision capabilities into large language models | 311 |
 jesolem/pcv | A Python module for computer vision tasks | 1,921 |
 pistony/residualattentionnetwork | A Gluon implementation of Residual Attention Network for image classification tasks | 107 |
 boyuanjiang/context_encoder_pytorch | A PyTorch implementation of a deep learning model for inpainting images using contextual information | 365 |