Loovelj / CV_attention

pytorch-version implementation codes of some CV attention models

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

About

pytorch-version implementation codes of some CV attention models


Languages

Language:Python 100.0%