├── .DS_Store ├── .gitignore ├── LICENSE ├── README.md ├── SUPPORT.md ├── config.py ├── configs └── smt │ ├── smt_base_224.yaml │ ├── smt_large_224.yaml │ ├── smt_large_224_22k.yaml │ ├── smt_large_224_22kto1k_finetune.yaml │ ├── smt_large_384_22kto1k_finetune.yaml │ ├── smt_small_224.yaml │ └── smt_tiny_224.yaml ├── data ├── __init__.py ├── build.py ├── cached_image_folder.py ├── imagenet22k_dataset.py ├── map22kto1k.txt ├── samplers.py └── zipreader.py ├── detection ├── .DS_Store ├── configs │ ├── atss_ressa_mstrain_3x_coco.py │ ├── atss_smt_b_fpn.py │ ├── atss_smt_s_fpn.py │ ├── cascade_mask_rcnn_ressa_mstrain_480-800_giou_4conv1f_adamw_3x_coco.py │ ├── cascade_mask_rcnn_smt_b_fpn.py │ ├── cascade_mask_rcnn_smt_s_fpn.py │ ├── mask_rcnn_ResSA_fpn_1x_coco.py │ ├── mask_rcnn_smt_b_fpn.py │ ├── mask_rcnn_smt_b_fpn_mstrain-poly_3x_coco.py │ ├── mask_rcnn_smt_s_fpn.py │ ├── retinanet_ressa_fpn_mstrain_480-800_3x_coco.py │ ├── retinanet_smt_b_fpn.py │ ├── retinanet_smt_s_fpn.py │ ├── sparse_rcnn_ressa_mstrain_3x_coco.py │ ├── sparse_rcnn_smt_b_fpn.py │ └── sparse_rcnn_smt_s_fpn.py └── mmdet │ └── models │ ├── backbones │ ├── __init__.py │ └── smt.py │ └── utils │ ├── __init__.py │ └── ckpt_convert.py ├── figures └── teaser.png ├── logger.py ├── lr_scheduler.py ├── main.py ├── models ├── __init__.py ├── build.py ├── smt.py └── swin_transformer.py ├── optimizer.py ├── segmentation ├── configs │ ├── upernet_smt_512x512_160k_ade20k.py │ ├── upernet_smt_b.py │ └── upernet_smt_s.py └── mmseg │ └── models │ ├── backbones │ ├── __init__.py │ └── smt.py │ └── utils │ ├── __init__.py │ └── ckpt_convert.py ├── utils.py └── visualiza ├── vis.ipynb └── vis_images ├── .DS_Store ├── 1.JPEG ├── 2.JPEG ├── 3.JPEG ├── 4.JPEG ├── 5.JPEG └── 6.JPEG /.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/.DS_Store -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/README.md -------------------------------------------------------------------------------- /SUPPORT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/SUPPORT.md -------------------------------------------------------------------------------- /config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/config.py -------------------------------------------------------------------------------- /configs/smt/smt_base_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/configs/smt/smt_base_224.yaml -------------------------------------------------------------------------------- /configs/smt/smt_large_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/configs/smt/smt_large_224.yaml -------------------------------------------------------------------------------- /configs/smt/smt_large_224_22k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/configs/smt/smt_large_224_22k.yaml -------------------------------------------------------------------------------- /configs/smt/smt_large_224_22kto1k_finetune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/configs/smt/smt_large_224_22kto1k_finetune.yaml -------------------------------------------------------------------------------- /configs/smt/smt_large_384_22kto1k_finetune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/configs/smt/smt_large_384_22kto1k_finetune.yaml -------------------------------------------------------------------------------- /configs/smt/smt_small_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/configs/smt/smt_small_224.yaml -------------------------------------------------------------------------------- /configs/smt/smt_tiny_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/configs/smt/smt_tiny_224.yaml -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/data/build.py -------------------------------------------------------------------------------- /data/cached_image_folder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/data/cached_image_folder.py -------------------------------------------------------------------------------- /data/imagenet22k_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/data/imagenet22k_dataset.py -------------------------------------------------------------------------------- /data/map22kto1k.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/data/map22kto1k.txt -------------------------------------------------------------------------------- /data/samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/data/samplers.py -------------------------------------------------------------------------------- /data/zipreader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/data/zipreader.py -------------------------------------------------------------------------------- /detection/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/detection/.DS_Store -------------------------------------------------------------------------------- /detection/configs/atss_ressa_mstrain_3x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/detection/configs/atss_ressa_mstrain_3x_coco.py -------------------------------------------------------------------------------- /detection/configs/atss_smt_b_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/detection/configs/atss_smt_b_fpn.py -------------------------------------------------------------------------------- /detection/configs/atss_smt_s_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/detection/configs/atss_smt_s_fpn.py -------------------------------------------------------------------------------- /detection/configs/cascade_mask_rcnn_ressa_mstrain_480-800_giou_4conv1f_adamw_3x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/detection/configs/cascade_mask_rcnn_ressa_mstrain_480-800_giou_4conv1f_adamw_3x_coco.py -------------------------------------------------------------------------------- /detection/configs/cascade_mask_rcnn_smt_b_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/detection/configs/cascade_mask_rcnn_smt_b_fpn.py -------------------------------------------------------------------------------- /detection/configs/cascade_mask_rcnn_smt_s_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/detection/configs/cascade_mask_rcnn_smt_s_fpn.py -------------------------------------------------------------------------------- /detection/configs/mask_rcnn_ResSA_fpn_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/detection/configs/mask_rcnn_ResSA_fpn_1x_coco.py -------------------------------------------------------------------------------- /detection/configs/mask_rcnn_smt_b_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/detection/configs/mask_rcnn_smt_b_fpn.py -------------------------------------------------------------------------------- /detection/configs/mask_rcnn_smt_b_fpn_mstrain-poly_3x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/detection/configs/mask_rcnn_smt_b_fpn_mstrain-poly_3x_coco.py -------------------------------------------------------------------------------- /detection/configs/mask_rcnn_smt_s_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/detection/configs/mask_rcnn_smt_s_fpn.py -------------------------------------------------------------------------------- /detection/configs/retinanet_ressa_fpn_mstrain_480-800_3x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/detection/configs/retinanet_ressa_fpn_mstrain_480-800_3x_coco.py -------------------------------------------------------------------------------- /detection/configs/retinanet_smt_b_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/detection/configs/retinanet_smt_b_fpn.py -------------------------------------------------------------------------------- /detection/configs/retinanet_smt_s_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/detection/configs/retinanet_smt_s_fpn.py -------------------------------------------------------------------------------- /detection/configs/sparse_rcnn_ressa_mstrain_3x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/detection/configs/sparse_rcnn_ressa_mstrain_3x_coco.py -------------------------------------------------------------------------------- /detection/configs/sparse_rcnn_smt_b_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/detection/configs/sparse_rcnn_smt_b_fpn.py -------------------------------------------------------------------------------- /detection/configs/sparse_rcnn_smt_s_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/detection/configs/sparse_rcnn_smt_s_fpn.py -------------------------------------------------------------------------------- /detection/mmdet/models/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/detection/mmdet/models/backbones/__init__.py -------------------------------------------------------------------------------- /detection/mmdet/models/backbones/smt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/detection/mmdet/models/backbones/smt.py -------------------------------------------------------------------------------- /detection/mmdet/models/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/detection/mmdet/models/utils/__init__.py -------------------------------------------------------------------------------- /detection/mmdet/models/utils/ckpt_convert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/detection/mmdet/models/utils/ckpt_convert.py -------------------------------------------------------------------------------- /figures/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/figures/teaser.png -------------------------------------------------------------------------------- /logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/logger.py -------------------------------------------------------------------------------- /lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/lr_scheduler.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/main.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/models/build.py -------------------------------------------------------------------------------- /models/smt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/models/smt.py -------------------------------------------------------------------------------- /models/swin_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/models/swin_transformer.py -------------------------------------------------------------------------------- /optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/optimizer.py -------------------------------------------------------------------------------- /segmentation/configs/upernet_smt_512x512_160k_ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/segmentation/configs/upernet_smt_512x512_160k_ade20k.py -------------------------------------------------------------------------------- /segmentation/configs/upernet_smt_b.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/segmentation/configs/upernet_smt_b.py -------------------------------------------------------------------------------- /segmentation/configs/upernet_smt_s.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/segmentation/configs/upernet_smt_s.py -------------------------------------------------------------------------------- /segmentation/mmseg/models/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/segmentation/mmseg/models/backbones/__init__.py -------------------------------------------------------------------------------- /segmentation/mmseg/models/backbones/smt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/segmentation/mmseg/models/backbones/smt.py -------------------------------------------------------------------------------- /segmentation/mmseg/models/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/segmentation/mmseg/models/utils/__init__.py -------------------------------------------------------------------------------- /segmentation/mmseg/models/utils/ckpt_convert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/segmentation/mmseg/models/utils/ckpt_convert.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/utils.py -------------------------------------------------------------------------------- /visualiza/vis.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/visualiza/vis.ipynb -------------------------------------------------------------------------------- /visualiza/vis_images/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/visualiza/vis_images/.DS_Store -------------------------------------------------------------------------------- /visualiza/vis_images/1.JPEG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/visualiza/vis_images/1.JPEG -------------------------------------------------------------------------------- /visualiza/vis_images/2.JPEG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/visualiza/vis_images/2.JPEG -------------------------------------------------------------------------------- /visualiza/vis_images/3.JPEG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/visualiza/vis_images/3.JPEG -------------------------------------------------------------------------------- /visualiza/vis_images/4.JPEG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/visualiza/vis_images/4.JPEG -------------------------------------------------------------------------------- /visualiza/vis_images/5.JPEG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/visualiza/vis_images/5.JPEG -------------------------------------------------------------------------------- /visualiza/vis_images/6.JPEG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AFeng-x/SMT/HEAD/visualiza/vis_images/6.JPEG --------------------------------------------------------------------------------