Keras implementation of CBAM; choose it for Keras/TensorFlow projects.
CBAM implementation on Keras
- 367
- Python
- MIT
computer-vision attention mechanisms (PyTorch)
A PyTorch library of many attention mechanisms (external attention, SE, CBAM, ECA, etc.) plus MLP/re-parameterization/convolution building blocks for studying vision papers.
Solves the same problem with a different architecture or at a different layer. Expect to rewrite the integration.
Keras implementation of CBAM; choose it for Keras/TensorFlow projects.
CBAM implementation on Keras
Keras SE block implementation; choose it for Keras-based projects needing squeeze-and-excitation.
Implementation of Squeeze and Excitation Networks in Keras
Original Caffe/CUDA SE-Net implementation; choose it if your stack is Caffe rather than PyTorch.
Squeeze-and-Excitation Networks
TensorFlow SE-Net implementation; choose it if you need SE blocks in TensorFlow.
Simple Tensorflow implementation of "Squeeze and Excitation Networks" using Cifar10 (ResNeXt, Inception-v4, Inception-resnet-v2)
An earlier or more limited way to do the job. Still the right call when you need something small, proven, or CPU-only.
Unmaintained PyTorch collection of classic attention modules; choose it for a smaller, self-contained codebase.
Self-attention、Non-local、SE、SK、CBAM、DANet
Official PyTorch CBAM/BAM source; choose it when you need the canonical implementations of these two modules.
Official PyTorch code for "BAM: Bottleneck Attention Module (BMVC2018)" and "CBAM: Convolutional Block Attention Module (ECCV2018)"
Similar stale PyTorch collection with ViT/MLP variants; choose it if you want a broader transformer-style module set.
🦖Pytorch implementation of popular Attention Mechanisms, Vision Transformers, MLP-Like models and CNNs.🔥🔥🔥
Standalone PyTorch CBAM layer; choose it when you only need a simple CBAM module.
Non-official implement of Paper:CBAM: Convolutional Block Attention Module
Older PyTorch self-attention collection for CV; choose it for a compact educational set.
Implementation of various self-attention mechanisms focused on computer vision. Ongoing repository.
PyTorch SE block with 2D/3D support; choose it if your model needs volumetric SE blocks.
PyTorch Implementation of 2D and 3D 'squeeze and excitation' blocks for Fully Convolutional Neural Networks
Official PyTorch ECA module code; choose it when you need just the ECA attention layer.
Code for ECA-Net: Efficient Channel Attention for Deep Convolutional Neural Networks
These projects were analysed and named External-Attention-pytorch among their alternatives. The relationship is not symmetric — how External-Attention-pytorch rates them is a separate judgement, made when External-Attention-pytorch is analysed in its own right.
calls External-Attention-pytorch “PyTorch implementations of attention and MLP mechanisms; choose for a maintained, focused attention library instead of numpy-ml's minimal attention.”
Machine learning, in numpy