Three self-attention modules based on three papers are implemented by PyTorch, which are:
Squeeze-and-Excitation Networks (resnet_se.py
),
CBAM: Convolutional Block Attention Module (resnet_cbam.py
), and
BAM: Bottleneck Attention Module (resnet_bam.py
),
with the baseline being ResNet
, proposed in
Deep Residual Learning for Image Recognition (resnet_base.py
).
-
Notifications
You must be signed in to change notification settings - Fork 2
Everglow0214/Self-attention-module
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
Homework #1 for EE898 (Advanced Topics in Deep Learning for Robotics and Computer Vision)
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published