← all repositories

luuuyi/CBAM.PyTorch

PyTorch implementation of CBAM (Convolutional Block Attention Module), an attention mechanism that refines CNN feature maps through sequential channel and spatial attention.

1.5k stars Python Computer VisionML Frameworks
CBAM.PyTorch
Not currently ranked — collecting fresh signals.
star history

This repository provides a non-official PyTorch reimplementation of the CBAM paper from ECCV 2018. CBAM is a plug-and-play attention module that can be integrated into any CNN architecture to adaptively refine intermediate feature maps. The implementation includes channel attention and spatial attention sub-modules applied sequentially at each convolutional block. Results are validated on ImageNet-1K using ResNet50, showing improved top-1 accuracy over baseline.

Frequently asked

What is luuuyi/CBAM.PyTorch?
PyTorch implementation of CBAM (Convolutional Block Attention Module), an attention mechanism that refines CNN feature maps through sequential channel and spatial attention.
Is CBAM.PyTorch open source?
Yes — luuuyi/CBAM.PyTorch is an open-source project tracked on heatdrop.
What language is CBAM.PyTorch written in?
luuuyi/CBAM.PyTorch is primarily written in Python.
How popular is CBAM.PyTorch?
luuuyi/CBAM.PyTorch has 1.5k stars on GitHub.
Where can I find CBAM.PyTorch?
luuuyi/CBAM.PyTorch is on GitHub at https://github.com/luuuyi/CBAM.PyTorch.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.