├── .gitignore ├── README.md ├── checkpoint.py ├── figures ├── attentions.png └── training_curves.png ├── main_cifar.py ├── main_imagenet.py ├── mmdetection ├── configs │ ├── _base_ │ │ ├── models │ │ │ ├── faster_rcnn_r50simam_fpn.py │ │ │ └── mask_rcnn_r50simam_fpn.py │ │ └── schedules │ │ │ └── schedule_1x_lr0.01.py │ ├── faster_rcnn │ │ ├── faster_rcnn_r101simam_fpn_1x_coco.py │ │ └── faster_rcnn_r50simam_fpn_1x_coco.py │ └── mask_rcnn │ │ ├── mask_rcnn_r101simam_fpn_1x_coco.py │ │ └── mask_rcnn_r50simam_fpn_1x_coco.py └── mmdet │ └── models │ └── backbones │ ├── attentions │ └── simam_module.py │ └── resnet_simam.py ├── networks ├── __init__.py ├── __pycache__ │ └── __init__.cpython-38.pyc ├── attentions │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-38.pyc │ │ ├── se_module.cpython-38.pyc │ │ └── simam_module.cpython-38.pyc │ ├── cbam_module.py │ ├── se_module.py │ └── simam_module.py ├── cifar │ ├── __init__.py │ ├── __pycache__ │ │ ├── __init__.cpython-38.pyc │ │ ├── block.cpython-38.pyc │ │ ├── mobilenetv2.cpython-38.pyc │ │ ├── preresnet.cpython-38.pyc │ │ ├── resnet.cpython-38.pyc │ │ └── wideresnet.cpython-38.pyc │ ├── block.py │ ├── mobilenetv2.py │ ├── preresnet.py │ ├── resnet.py │ └── wideresnet.py └── imagenet │ ├── __init__.py │ ├── __pycache__ │ ├── __init__.cpython-38.pyc │ ├── mobilenetv2.cpython-38.pyc │ └── resnet.cpython-38.pyc │ ├── mobilenetv2.py │ └── resnet.py └── util.py /.gitignore: -------------------------------------------------------------------------------- 1 | pretrained 2 | SimAM-previous.rar -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/README.md -------------------------------------------------------------------------------- /checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/checkpoint.py -------------------------------------------------------------------------------- /figures/attentions.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/figures/attentions.png -------------------------------------------------------------------------------- /figures/training_curves.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/figures/training_curves.png -------------------------------------------------------------------------------- /main_cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/main_cifar.py -------------------------------------------------------------------------------- /main_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/main_imagenet.py -------------------------------------------------------------------------------- /mmdetection/configs/_base_/models/faster_rcnn_r50simam_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/mmdetection/configs/_base_/models/faster_rcnn_r50simam_fpn.py -------------------------------------------------------------------------------- /mmdetection/configs/_base_/models/mask_rcnn_r50simam_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/mmdetection/configs/_base_/models/mask_rcnn_r50simam_fpn.py -------------------------------------------------------------------------------- /mmdetection/configs/_base_/schedules/schedule_1x_lr0.01.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/mmdetection/configs/_base_/schedules/schedule_1x_lr0.01.py -------------------------------------------------------------------------------- /mmdetection/configs/faster_rcnn/faster_rcnn_r101simam_fpn_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/mmdetection/configs/faster_rcnn/faster_rcnn_r101simam_fpn_1x_coco.py -------------------------------------------------------------------------------- /mmdetection/configs/faster_rcnn/faster_rcnn_r50simam_fpn_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/mmdetection/configs/faster_rcnn/faster_rcnn_r50simam_fpn_1x_coco.py -------------------------------------------------------------------------------- /mmdetection/configs/mask_rcnn/mask_rcnn_r101simam_fpn_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/mmdetection/configs/mask_rcnn/mask_rcnn_r101simam_fpn_1x_coco.py -------------------------------------------------------------------------------- /mmdetection/configs/mask_rcnn/mask_rcnn_r50simam_fpn_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/mmdetection/configs/mask_rcnn/mask_rcnn_r50simam_fpn_1x_coco.py -------------------------------------------------------------------------------- /mmdetection/mmdet/models/backbones/attentions/simam_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/mmdetection/mmdet/models/backbones/attentions/simam_module.py -------------------------------------------------------------------------------- /mmdetection/mmdet/models/backbones/resnet_simam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/mmdetection/mmdet/models/backbones/resnet_simam.py -------------------------------------------------------------------------------- /networks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/networks/__init__.py -------------------------------------------------------------------------------- /networks/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/networks/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /networks/attentions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/networks/attentions/__init__.py -------------------------------------------------------------------------------- /networks/attentions/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/networks/attentions/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /networks/attentions/__pycache__/se_module.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/networks/attentions/__pycache__/se_module.cpython-38.pyc -------------------------------------------------------------------------------- /networks/attentions/__pycache__/simam_module.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/networks/attentions/__pycache__/simam_module.cpython-38.pyc -------------------------------------------------------------------------------- /networks/attentions/cbam_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/networks/attentions/cbam_module.py -------------------------------------------------------------------------------- /networks/attentions/se_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/networks/attentions/se_module.py -------------------------------------------------------------------------------- /networks/attentions/simam_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/networks/attentions/simam_module.py -------------------------------------------------------------------------------- /networks/cifar/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/networks/cifar/__init__.py -------------------------------------------------------------------------------- /networks/cifar/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/networks/cifar/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /networks/cifar/__pycache__/block.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/networks/cifar/__pycache__/block.cpython-38.pyc -------------------------------------------------------------------------------- /networks/cifar/__pycache__/mobilenetv2.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/networks/cifar/__pycache__/mobilenetv2.cpython-38.pyc -------------------------------------------------------------------------------- /networks/cifar/__pycache__/preresnet.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/networks/cifar/__pycache__/preresnet.cpython-38.pyc -------------------------------------------------------------------------------- /networks/cifar/__pycache__/resnet.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/networks/cifar/__pycache__/resnet.cpython-38.pyc -------------------------------------------------------------------------------- /networks/cifar/__pycache__/wideresnet.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/networks/cifar/__pycache__/wideresnet.cpython-38.pyc -------------------------------------------------------------------------------- /networks/cifar/block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/networks/cifar/block.py -------------------------------------------------------------------------------- /networks/cifar/mobilenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/networks/cifar/mobilenetv2.py -------------------------------------------------------------------------------- /networks/cifar/preresnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/networks/cifar/preresnet.py -------------------------------------------------------------------------------- /networks/cifar/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/networks/cifar/resnet.py -------------------------------------------------------------------------------- /networks/cifar/wideresnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/networks/cifar/wideresnet.py -------------------------------------------------------------------------------- /networks/imagenet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/networks/imagenet/__init__.py -------------------------------------------------------------------------------- /networks/imagenet/__pycache__/__init__.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/networks/imagenet/__pycache__/__init__.cpython-38.pyc -------------------------------------------------------------------------------- /networks/imagenet/__pycache__/mobilenetv2.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/networks/imagenet/__pycache__/mobilenetv2.cpython-38.pyc -------------------------------------------------------------------------------- /networks/imagenet/__pycache__/resnet.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/networks/imagenet/__pycache__/resnet.cpython-38.pyc -------------------------------------------------------------------------------- /networks/imagenet/mobilenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/networks/imagenet/mobilenetv2.py -------------------------------------------------------------------------------- /networks/imagenet/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/networks/imagenet/resnet.py -------------------------------------------------------------------------------- /util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZjjConan/SimAM/HEAD/util.py --------------------------------------------------------------------------------