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 | 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 |