Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/MenghaoGuo/Awesome-Vision-Attentions
/ types & classes
Types & classes
57 in github.com/MenghaoGuo/Awesome-Vision-Attentions
⨍
Functions
164
◇
Types & classes
57
↓ 3 callers
Class
AttentionGate
code/channel_spatial_attentions/triplet_attention.py:55
↓ 2 callers
Class
MultiHeadedAttention
Take in model size and number of heads.
code/spatial_temporal_attentions/dstt_module.py:25
↓ 2 callers
Class
small_cell
code/channel_attentions/dia_module.py:6
↓ 1 callers
Class
Attention
Compute 'Scaled Dot Product Attention
code/spatial_temporal_attentions/dstt_module.py:7
↓ 1 callers
Class
Attention
code/channel_attentions/dia_module.py:62
↓ 1 callers
Class
AttentionModule
code/channel_spatial_attentions/lka.py:6
↓ 1 callers
Class
AugmentedConv
code/spatial_attentions/attention_augmented_module.py:6
↓ 1 callers
Class
BAM
code/channel_spatial_attentions/bam.py:55
↓ 1 callers
Class
BasicConv
code/channel_spatial_attentions/cbam.py:5
↓ 1 callers
Class
BasicConv
code/channel_spatial_attentions/triplet_attention.py:6
↓ 1 callers
Class
CAM_Module
Channel attention module
code/channel_spatial_attentions/danet.py:88
↓ 1 callers
Class
CBAM
code/channel_spatial_attentions/cbam.py:81
↓ 1 callers
Class
ChannelGate
code/channel_spatial_attentions/bam.py:11
↓ 1 callers
Class
ChannelGate
code/channel_spatial_attentions/cbam.py:24
↓ 1 callers
Class
ChannelPool
code/channel_spatial_attentions/cbam.py:54
↓ 1 callers
Class
CoordAtt
code/channel_spatial_attentions/coordatt_module.py:24
↓ 1 callers
Class
Covpool
code/channel_attentions/soca_module.py:6
↓ 1 callers
Class
DANetHead
code/channel_spatial_attentions/danet.py:5
↓ 1 callers
Class
DoubleAtten
code/spatial_attentions/doub_attention.py:6
↓ 1 callers
Class
Dynamic_conv2d
code/branch_attentions/dynamic_conv.py:39
↓ 1 callers
Class
ECALayer
Constructs a ECA module. Args: k_size: Adaptive selection of kernel size
code/channel_attentions/eca_module.py:5
↓ 1 callers
Class
EfficientAttention
code/spatial_attentions/segformer_module.py:6
↓ 1 callers
Class
EncModule
code/channel_attentions/enc_module.py:57
↓ 1 callers
Class
Encoding
code/channel_attentions/enc_module.py:6
↓ 1 callers
Class
External_attention
Arguments: c (int): The input and output channel number.
code/spatial_attentions/external_attention.py:6
↓ 1 callers
Class
Flatten
code/channel_spatial_attentions/bam.py:6
↓ 1 callers
Class
GCT
code/channel_attentions/gct_module.py:6
↓ 1 callers
Class
GlobalContextBlock
code/spatial_attentions/gc_module.py:5
↓ 1 callers
Class
Hamburger
code/spatial_attentions/hamnet.py:53
↓ 1 callers
Class
Highway
code/branch_attentions/highway.py:5
↓ 1 callers
Class
LSTMCell
code/channel_attentions/dia_module.py:18
↓ 1 callers
Class
MHSA
code/spatial_attentions/mhsa.py:5
↓ 1 callers
Class
MultiSpectralAttentionLayer
code/channel_attentions/fcanet.py:38
↓ 1 callers
Class
MultiSpectralDCTLayer
Generate dct filters
code/channel_attentions/fcanet.py:76
↓ 1 callers
Class
NMF
code/spatial_attentions/hamnet.py:12
↓ 1 callers
Class
PAM_Module
Position attention module
code/channel_spatial_attentions/danet.py:47
↓ 1 callers
Class
RadixSoftmax
code/branch_attentions/resnest_module.py:15
↓ 1 callers
Class
SA_Layer
code/spatial_attentions/offset_module.py:7
↓ 1 callers
Class
SCConv
code/channel_spatial_attentions/scnet.py:6
↓ 1 callers
Class
SELayer
code/channel_attentions/se_module.py:5
↓ 1 callers
Class
SKModule
code/branch_attentions/sk_module.py:6
↓ 1 callers
Class
SOCA
code/channel_attentions/soca_module.py:107
↓ 1 callers
Class
STN
code/spatial_attentions/stn.py:75
↓ 1 callers
Class
SelfAttention
self attention module
code/spatial_attentions/self_attention.py:5
↓ 1 callers
Class
SpatialAttention
code/channel_spatial_attentions/lka.py:23
↓ 1 callers
Class
SpatialGate
code/channel_spatial_attentions/bam.py:33
↓ 1 callers
Class
SpatialGate
code/channel_spatial_attentions/cbam.py:65
↓ 1 callers
Class
SplitAttn
Split-Attention (aka Splat)
code/branch_attentions/resnest_module.py:32
↓ 1 callers
Class
Sqrtm
code/channel_attentions/soca_module.py:36
↓ 1 callers
Class
StripPooling
Reference:
code/channel_spatial_attentions/strip_pooling_module.py:6
↓ 1 callers
Class
TripletAttention
code/channel_spatial_attentions/triplet_attention.py:71
↓ 1 callers
Class
ZPool
code/channel_spatial_attentions/triplet_attention.py:48
↓ 1 callers
Class
attention2d
code/branch_attentions/dynamic_conv.py:6
↓ 1 callers
Class
h_sigmoid
code/channel_spatial_attentions/coordatt_module.py:6
↓ 1 callers
Class
h_swish
code/channel_spatial_attentions/coordatt_module.py:15
↓ 1 callers
Class
simam_module
code/channel_spatial_attentions/simam_module.py:6
Class
OCRHead
code/spatial_attentions/ocr.py:5