├── .gitignore ├── LICENSE ├── README.md ├── configs ├── __init__.py ├── config.py ├── msg_base_p4_win7_224.yaml ├── msg_small_p4_win7_224.yaml └── msg_tiny_p4_win7_224.yaml ├── data ├── __init__.py ├── build.py └── samplers.py ├── imgs ├── arch.png ├── block.png ├── coco.png └── imagenet.png ├── logger.py ├── lr_scheduler.py ├── main.py ├── mindspore ├── README.md ├── __init__.py ├── build.py ├── configs │ ├── __init__.py │ ├── config.py │ ├── msg_base_p4_win7_224.yaml │ ├── msg_small_p4_win7_224.yaml │ └── msg_tiny_p4_win7_224.yaml ├── msg_transformer_mindspore.py ├── test.py └── timm_ms.py ├── models ├── __init__.py ├── build.py └── msg_transformer.py ├── optimizer.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/MSG-Transformer/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/MSG-Transformer/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/MSG-Transformer/HEAD/README.md -------------------------------------------------------------------------------- /configs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /configs/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/MSG-Transformer/HEAD/configs/config.py -------------------------------------------------------------------------------- /configs/msg_base_p4_win7_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/MSG-Transformer/HEAD/configs/msg_base_p4_win7_224.yaml -------------------------------------------------------------------------------- /configs/msg_small_p4_win7_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/MSG-Transformer/HEAD/configs/msg_small_p4_win7_224.yaml -------------------------------------------------------------------------------- /configs/msg_tiny_p4_win7_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/MSG-Transformer/HEAD/configs/msg_tiny_p4_win7_224.yaml -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/MSG-Transformer/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/MSG-Transformer/HEAD/data/build.py -------------------------------------------------------------------------------- /data/samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/MSG-Transformer/HEAD/data/samplers.py -------------------------------------------------------------------------------- /imgs/arch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/MSG-Transformer/HEAD/imgs/arch.png -------------------------------------------------------------------------------- /imgs/block.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/MSG-Transformer/HEAD/imgs/block.png -------------------------------------------------------------------------------- /imgs/coco.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/MSG-Transformer/HEAD/imgs/coco.png -------------------------------------------------------------------------------- /imgs/imagenet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/MSG-Transformer/HEAD/imgs/imagenet.png -------------------------------------------------------------------------------- /logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/MSG-Transformer/HEAD/logger.py -------------------------------------------------------------------------------- /lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/MSG-Transformer/HEAD/lr_scheduler.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/MSG-Transformer/HEAD/main.py -------------------------------------------------------------------------------- /mindspore/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/MSG-Transformer/HEAD/mindspore/README.md -------------------------------------------------------------------------------- /mindspore/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/MSG-Transformer/HEAD/mindspore/__init__.py -------------------------------------------------------------------------------- /mindspore/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/MSG-Transformer/HEAD/mindspore/build.py -------------------------------------------------------------------------------- /mindspore/configs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/MSG-Transformer/HEAD/mindspore/configs/__init__.py -------------------------------------------------------------------------------- /mindspore/configs/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/MSG-Transformer/HEAD/mindspore/configs/config.py -------------------------------------------------------------------------------- /mindspore/configs/msg_base_p4_win7_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/MSG-Transformer/HEAD/mindspore/configs/msg_base_p4_win7_224.yaml -------------------------------------------------------------------------------- /mindspore/configs/msg_small_p4_win7_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/MSG-Transformer/HEAD/mindspore/configs/msg_small_p4_win7_224.yaml -------------------------------------------------------------------------------- /mindspore/configs/msg_tiny_p4_win7_224.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/MSG-Transformer/HEAD/mindspore/configs/msg_tiny_p4_win7_224.yaml -------------------------------------------------------------------------------- /mindspore/msg_transformer_mindspore.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/MSG-Transformer/HEAD/mindspore/msg_transformer_mindspore.py -------------------------------------------------------------------------------- /mindspore/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/MSG-Transformer/HEAD/mindspore/test.py -------------------------------------------------------------------------------- /mindspore/timm_ms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/MSG-Transformer/HEAD/mindspore/timm_ms.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/MSG-Transformer/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/MSG-Transformer/HEAD/models/build.py -------------------------------------------------------------------------------- /models/msg_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/MSG-Transformer/HEAD/models/msg_transformer.py -------------------------------------------------------------------------------- /optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/MSG-Transformer/HEAD/optimizer.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hustvl/MSG-Transformer/HEAD/utils.py --------------------------------------------------------------------------------