CV_attention icon indicating copy to clipboard operation
CV_attention copied to clipboard

pytorch-version implementation codes of some CV attention models

CV Attention Models Reproduction

pytorch-version implementation codes of some CV attention models (some models are also provided with paddlepaddle-version)

All attention models here are plug-and-play blocks

Keep updating, just enjoy using them!

model zoo

model paper link
SEnet Squeeze-and-Excitation Networks
Non-local Non-local Neural Networks
CBAM CBAM: Convolutional Block Attention Module
GCnet GCNet: Non-local Networks Meet Squeeze-Excitation Networks and Beyond
DAnet Dual Attention Network for Scene Segmentation(CVPR2019)
updating updating