MCPcopy Create free account

hub / github.com/ZhugeKongan/Attention-mechanism-implementation / types & classes

Types & classes43 in github.com/ZhugeKongan/Attention-mechanism-implementation

⨍ Functions 81◇ Types & classes 43
↓ 4 callersClassAverageMeter
Computes and stores the average and current value
train.py:150
↓ 3 callersClassSpatialGate
models/TripletAttention.py:17
↓ 2 callersClassSpatialPyramidPooling
models/ANN.py:12
↓ 1 callersClassAFNB
models/ANN.py:58
↓ 1 callersClassCBAM
models/CBAM_blocks.py:48
↓ 1 callersClassChannelAttention
models/CBAM_blocks.py:7
↓ 1 callersClassChannelPool
models/TripletAttention.py:12
↓ 1 callersClassECA
Constructs a ECA module. Args: channel: Number of channels of the input feature map k_size: Adaptive selection of kernel size
models/SK_block.py:97
↓ 1 callersClassGlobalContextBlock
models/GlobalContextBlock.py:6
↓ 1 callersClassNONLocalBlock1D
models/non_local_dot_product.py:93
↓ 1 callersClassNONLocalBlock1D
models/non_local_embedded_gaussian.py:99
↓ 1 callersClassNONLocalBlock1D
models/non_local_gaussian.py:95
↓ 1 callersClassNONLocalBlock1D
models/non_local_concatenation.py:109
↓ 1 callersClassNONLocalBlock2D
models/non_local_dot_product.py:101
↓ 1 callersClassNONLocalBlock2D
models/non_local_embedded_gaussian.py:107
↓ 1 callersClassNONLocalBlock2D
models/non_local_gaussian.py:103
↓ 1 callersClassNONLocalBlock2D
models/non_local_concatenation.py:117
↓ 1 callersClassNONLocalBlock3D
models/non_local_dot_product.py:109
↓ 1 callersClassNONLocalBlock3D
models/non_local_embedded_gaussian.py:115
↓ 1 callersClassNONLocalBlock3D
models/non_local_gaussian.py:111
↓ 1 callersClassNONLocalBlock3D
models/non_local_concatenation.py:125
↓ 1 callersClassNonLocalBlock
models/NonLocalBlock.py:6
↓ 1 callersClassResNet
train.py:78
↓ 1 callersClassSKConv
models/SKNet.py:5
↓ 1 callersClassSpatialAttention
models/CBAM_blocks.py:30
↓ 1 callersClassTripletAttention
models/TripletAttention.py:33
↓ 1 callersClassWarmUpLR
warmup_training learning rate scheduler Args: optimizer: optimzier(e.g. SGD) total_iters: totoal_iters of warmup phase
train.py:133
↓ 1 callersClass_ChannelAttentionModule
Channel attention module
models/DaNet.py:28
↓ 1 callersClass_PositionAttentionModule
Position attention module
models/DaNet.py:5
↓ 1 callersClasscSE_Module
models/SEvariants.py:6
↓ 1 callersClasssSE_Module
models/SEvariants.py:23
↓ 1 callersClassscSE_Module
models/SEvariants.py:35
ClassAPNB
models/ANN.py:27
ClassBasicBlock
train.py:46
ClassSE
models/SE_block.py:11
ClassSF
models/SK_block.py:120
ClassSKConv
models/SK_block.py:3
Class_DAHead
models/DaNet.py:50
Class_NonLocalBlockND
models/non_local_dot_product.py:6
Class_NonLocalBlockND
models/non_local_embedded_gaussian.py:6
Class_NonLocalBlockND
models/non_local_gaussian.py:6
Class_NonLocalBlockND
models/non_local_concatenation.py:6
Classresidual_attention
models/residual_attention_blocks.py:6