attention-module
by Jongchan
Official PyTorch code for "BAM: Bottleneck Attention Module (BMVC2018)" and "CBAM: Convolutional Block Attention Module (ECCV2018)"
AI summary
Attention module library
Provides implementations of attention modules for computer vision tasks using PyTorch
- stars
- 2.1K
- forks
- 402
- watching
- 19
- awesome list
- 1
Similar projects
Found by comparing what the projects do, not just their names.
Attention module implementation
An implementation of the Bottleneck Attention Module in TensorFlow using attention mechanism
elbuco1/cbam107
Attention module
A deep learning project that develops and tests a novel attention mechanism called CBAM for image classification tasks.
Feature map refiner
PyTorch implementation of the CBAM module for refining feature maps in deep networks
Computer Vision Toolkit
A PyTorch toolbox for supporting research and development of domain adaptation, generalization, and semi-supervised learning methods in computer vision.
Attention module
A repository implementing a Keras-based architecture for improving neural network representation power with attention mechanisms and squeeze-and-excitation networks.
Image classifier
Implementing a deep learning framework for image classification using Residual Attention Network architecture
Semantic Segmentation Model
An implementation of a deep learning model using PyTorch for semantic segmentation tasks.
Image synthesis model
An implementation of semantic image synthesis via adversarial learning using PyTorch
Deep learning toolkit
A Python framework for building deep learning models with optimized encoding layers and batch normalization.
Object attention framework
A PyTorch implementation of an attention-guided inference network to focus on specific areas of objects in images
Deblurring model
A PyTorch implementation of an attention network for dynamic scene deblurring
Vision Language Integrator
Improves performance of vision language tasks by integrating computer vision capabilities into large language models
jesolem/pcv1.9K
CV library
A Python module for computer vision tasks
Attention network
A Gluon implementation of Residual Attention Network for image classification tasks
Image inpainter
A PyTorch implementation of a deep learning model for inpainting images using contextual information
Featured in 1 awesome list
Each link jumps to the spot where the list mentions attention-module.
