├── README.md ├── models ├── ANN.py ├── CBAM_blocks.py ├── CCNet.py ├── DaNet.py ├── GlobalContextBlock.py ├── NonLocalBlock.py ├── SE_block.py ├── SEvariants.py ├── SKNet.py ├── SK_block.py ├── TripletAttention.py ├── non_local_concatenation.py ├── non_local_dot_product.py ├── non_local_embedded_gaussian.py ├── non_local_gaussian.py └── residual_attention_blocks.py └── train.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhugeKongan/Attention-mechanism-implementation/HEAD/README.md -------------------------------------------------------------------------------- /models/ANN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhugeKongan/Attention-mechanism-implementation/HEAD/models/ANN.py -------------------------------------------------------------------------------- /models/CBAM_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhugeKongan/Attention-mechanism-implementation/HEAD/models/CBAM_blocks.py -------------------------------------------------------------------------------- /models/CCNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhugeKongan/Attention-mechanism-implementation/HEAD/models/CCNet.py -------------------------------------------------------------------------------- /models/DaNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhugeKongan/Attention-mechanism-implementation/HEAD/models/DaNet.py -------------------------------------------------------------------------------- /models/GlobalContextBlock.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhugeKongan/Attention-mechanism-implementation/HEAD/models/GlobalContextBlock.py -------------------------------------------------------------------------------- /models/NonLocalBlock.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhugeKongan/Attention-mechanism-implementation/HEAD/models/NonLocalBlock.py -------------------------------------------------------------------------------- /models/SE_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhugeKongan/Attention-mechanism-implementation/HEAD/models/SE_block.py -------------------------------------------------------------------------------- /models/SEvariants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhugeKongan/Attention-mechanism-implementation/HEAD/models/SEvariants.py -------------------------------------------------------------------------------- /models/SKNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhugeKongan/Attention-mechanism-implementation/HEAD/models/SKNet.py -------------------------------------------------------------------------------- /models/SK_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhugeKongan/Attention-mechanism-implementation/HEAD/models/SK_block.py -------------------------------------------------------------------------------- /models/TripletAttention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhugeKongan/Attention-mechanism-implementation/HEAD/models/TripletAttention.py -------------------------------------------------------------------------------- /models/non_local_concatenation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhugeKongan/Attention-mechanism-implementation/HEAD/models/non_local_concatenation.py -------------------------------------------------------------------------------- /models/non_local_dot_product.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhugeKongan/Attention-mechanism-implementation/HEAD/models/non_local_dot_product.py -------------------------------------------------------------------------------- /models/non_local_embedded_gaussian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhugeKongan/Attention-mechanism-implementation/HEAD/models/non_local_embedded_gaussian.py -------------------------------------------------------------------------------- /models/non_local_gaussian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhugeKongan/Attention-mechanism-implementation/HEAD/models/non_local_gaussian.py -------------------------------------------------------------------------------- /models/residual_attention_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhugeKongan/Attention-mechanism-implementation/HEAD/models/residual_attention_blocks.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhugeKongan/Attention-mechanism-implementation/HEAD/train.py --------------------------------------------------------------------------------