├── .gitattributes ├── .gitignore ├── ObjectDetection ├── configs │ ├── _base_ │ │ └── models │ │ │ ├── cascade_mask_rcnn_swin_visformer_fpn.py │ │ │ └── mask_rcnn_swin_visformer_fpn.py │ └── swin_visformer │ │ ├── cascade_mask_rcnn_swin_visformer_small_v2_mstrain_480-800_adamw_1x_coco.py │ │ ├── cascade_mask_rcnn_swin_visformer_small_v2_mstrain_480-800_adamw_3x_coco.py │ │ ├── mask_rcnn_swin_visformer_small_v2_adamw_1x_coco.py │ │ └── mask_rcnn_swin_visformer_small_v2_mstrain_480-800_adamw_3x_coco.py └── mmdet │ └── models │ └── backbones │ └── swin_visformer.py ├── README.md ├── compared_models ├── __init__.py ├── autoformer │ ├── AutoFormer-B.yaml │ ├── AutoFormer-S.yaml │ ├── AutoFormer-T.yaml │ ├── __init__.py │ ├── config.py │ ├── module │ │ ├── Linear_super.py │ │ ├── __init__.py │ │ ├── embedding_super.py │ │ ├── layernorm_super.py │ │ ├── multihead_super.py │ │ └── qkv_super.py │ ├── supernet_transformer.py │ └── utils.py ├── conformer.py ├── cswin.py ├── cvt │ ├── __init__.py │ ├── build.py │ ├── cls_cvt.py │ ├── config │ │ ├── __init__.py │ │ └── default.py │ ├── cvt-13-224x224.yaml │ ├── cvt-21-224x224.yaml │ └── registry.py ├── focal_transformer.py ├── pvt.py ├── pvt_v2.py ├── t2t_vit.py ├── token_performer.py ├── token_transformer.py └── transformer_block.py ├── datasets.py ├── engine.py ├── get_throughput.py ├── get_visformer_flops.py ├── main.py ├── mobilenetv2.py ├── models.py ├── requirements.txt ├── samplers.py ├── swin_models.py ├── utils.py └── weight_init.py /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/.gitattributes -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/.gitignore -------------------------------------------------------------------------------- /ObjectDetection/configs/_base_/models/cascade_mask_rcnn_swin_visformer_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/ObjectDetection/configs/_base_/models/cascade_mask_rcnn_swin_visformer_fpn.py -------------------------------------------------------------------------------- /ObjectDetection/configs/_base_/models/mask_rcnn_swin_visformer_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/ObjectDetection/configs/_base_/models/mask_rcnn_swin_visformer_fpn.py -------------------------------------------------------------------------------- /ObjectDetection/configs/swin_visformer/cascade_mask_rcnn_swin_visformer_small_v2_mstrain_480-800_adamw_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/ObjectDetection/configs/swin_visformer/cascade_mask_rcnn_swin_visformer_small_v2_mstrain_480-800_adamw_1x_coco.py -------------------------------------------------------------------------------- /ObjectDetection/configs/swin_visformer/cascade_mask_rcnn_swin_visformer_small_v2_mstrain_480-800_adamw_3x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/ObjectDetection/configs/swin_visformer/cascade_mask_rcnn_swin_visformer_small_v2_mstrain_480-800_adamw_3x_coco.py -------------------------------------------------------------------------------- /ObjectDetection/configs/swin_visformer/mask_rcnn_swin_visformer_small_v2_adamw_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/ObjectDetection/configs/swin_visformer/mask_rcnn_swin_visformer_small_v2_adamw_1x_coco.py -------------------------------------------------------------------------------- /ObjectDetection/configs/swin_visformer/mask_rcnn_swin_visformer_small_v2_mstrain_480-800_adamw_3x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/ObjectDetection/configs/swin_visformer/mask_rcnn_swin_visformer_small_v2_mstrain_480-800_adamw_3x_coco.py -------------------------------------------------------------------------------- /ObjectDetection/mmdet/models/backbones/swin_visformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/ObjectDetection/mmdet/models/backbones/swin_visformer.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/README.md -------------------------------------------------------------------------------- /compared_models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/compared_models/__init__.py -------------------------------------------------------------------------------- /compared_models/autoformer/AutoFormer-B.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/compared_models/autoformer/AutoFormer-B.yaml -------------------------------------------------------------------------------- /compared_models/autoformer/AutoFormer-S.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/compared_models/autoformer/AutoFormer-S.yaml -------------------------------------------------------------------------------- /compared_models/autoformer/AutoFormer-T.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/compared_models/autoformer/AutoFormer-T.yaml -------------------------------------------------------------------------------- /compared_models/autoformer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/compared_models/autoformer/__init__.py -------------------------------------------------------------------------------- /compared_models/autoformer/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/compared_models/autoformer/config.py -------------------------------------------------------------------------------- /compared_models/autoformer/module/Linear_super.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/compared_models/autoformer/module/Linear_super.py -------------------------------------------------------------------------------- /compared_models/autoformer/module/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /compared_models/autoformer/module/embedding_super.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/compared_models/autoformer/module/embedding_super.py -------------------------------------------------------------------------------- /compared_models/autoformer/module/layernorm_super.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/compared_models/autoformer/module/layernorm_super.py -------------------------------------------------------------------------------- /compared_models/autoformer/module/multihead_super.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/compared_models/autoformer/module/multihead_super.py -------------------------------------------------------------------------------- /compared_models/autoformer/module/qkv_super.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/compared_models/autoformer/module/qkv_super.py -------------------------------------------------------------------------------- /compared_models/autoformer/supernet_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/compared_models/autoformer/supernet_transformer.py -------------------------------------------------------------------------------- /compared_models/autoformer/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/compared_models/autoformer/utils.py -------------------------------------------------------------------------------- /compared_models/conformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/compared_models/conformer.py -------------------------------------------------------------------------------- /compared_models/cswin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/compared_models/cswin.py -------------------------------------------------------------------------------- /compared_models/cvt/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/compared_models/cvt/__init__.py -------------------------------------------------------------------------------- /compared_models/cvt/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/compared_models/cvt/build.py -------------------------------------------------------------------------------- /compared_models/cvt/cls_cvt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/compared_models/cvt/cls_cvt.py -------------------------------------------------------------------------------- /compared_models/cvt/config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/compared_models/cvt/config/__init__.py -------------------------------------------------------------------------------- /compared_models/cvt/config/default.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/compared_models/cvt/config/default.py -------------------------------------------------------------------------------- /compared_models/cvt/cvt-13-224x224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/compared_models/cvt/cvt-13-224x224.yaml -------------------------------------------------------------------------------- /compared_models/cvt/cvt-21-224x224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/compared_models/cvt/cvt-21-224x224.yaml -------------------------------------------------------------------------------- /compared_models/cvt/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/compared_models/cvt/registry.py -------------------------------------------------------------------------------- /compared_models/focal_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/compared_models/focal_transformer.py -------------------------------------------------------------------------------- /compared_models/pvt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/compared_models/pvt.py -------------------------------------------------------------------------------- /compared_models/pvt_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/compared_models/pvt_v2.py -------------------------------------------------------------------------------- /compared_models/t2t_vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/compared_models/t2t_vit.py -------------------------------------------------------------------------------- /compared_models/token_performer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/compared_models/token_performer.py -------------------------------------------------------------------------------- /compared_models/token_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/compared_models/token_transformer.py -------------------------------------------------------------------------------- /compared_models/transformer_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/compared_models/transformer_block.py -------------------------------------------------------------------------------- /datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/datasets.py -------------------------------------------------------------------------------- /engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/engine.py -------------------------------------------------------------------------------- /get_throughput.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/get_throughput.py -------------------------------------------------------------------------------- /get_visformer_flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/get_visformer_flops.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/main.py -------------------------------------------------------------------------------- /mobilenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/mobilenetv2.py -------------------------------------------------------------------------------- /models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/models.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/requirements.txt -------------------------------------------------------------------------------- /samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/samplers.py -------------------------------------------------------------------------------- /swin_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/swin_models.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/utils.py -------------------------------------------------------------------------------- /weight_init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/danczs/Visformer/HEAD/weight_init.py --------------------------------------------------------------------------------