├── .github ├── demo.png ├── distillation_position_in_d2.svg └── intro@mgd.light.svg ├── .gitignore ├── LICENSE ├── README.md ├── cls ├── README.md ├── main_base.py ├── main_mgd.py ├── models │ ├── __init__.py │ ├── mobilenetv1.py │ ├── mobilenetv2.py │ ├── resnet.py │ ├── shufflenetv2.py │ └── utils.py └── solver │ ├── __init__.py │ └── lr_scheduler.py ├── d2 ├── README.md ├── configs │ ├── retinanet_torchvision_R_18_FPN_1x.yaml │ └── retinanet_torchvision_R_50_FPN_1x.yaml ├── d2 │ ├── engine │ │ └── defaults.py │ └── modeling │ │ ├── backbone │ │ └── resnet.py │ │ └── meta_arch │ │ └── retinanet.py ├── mgd │ ├── __init__.py │ ├── builder.py │ └── config.py ├── scripts │ ├── convert-output-to-d2.py │ └── merge-models-into-one.py └── train_net.py ├── mgd ├── __init__.py ├── builder.py └── sampler.py └── unsup ├── README.md ├── main_moco_mgd.py ├── mgd ├── __init__.py └── builder.py ├── moco ├── __init__.py ├── builder.py └── loader.py └── models ├── __init__.py └── resnet.py /.github/demo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kaiyuyue/mgd/HEAD/.github/demo.png -------------------------------------------------------------------------------- /.github/distillation_position_in_d2.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kaiyuyue/mgd/HEAD/.github/distillation_position_in_d2.svg -------------------------------------------------------------------------------- /.github/intro@mgd.light.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kaiyuyue/mgd/HEAD/.github/intro@mgd.light.svg -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kaiyuyue/mgd/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kaiyuyue/mgd/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kaiyuyue/mgd/HEAD/README.md -------------------------------------------------------------------------------- /cls/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kaiyuyue/mgd/HEAD/cls/README.md -------------------------------------------------------------------------------- /cls/main_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kaiyuyue/mgd/HEAD/cls/main_base.py -------------------------------------------------------------------------------- /cls/main_mgd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kaiyuyue/mgd/HEAD/cls/main_mgd.py -------------------------------------------------------------------------------- /cls/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kaiyuyue/mgd/HEAD/cls/models/__init__.py -------------------------------------------------------------------------------- /cls/models/mobilenetv1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kaiyuyue/mgd/HEAD/cls/models/mobilenetv1.py -------------------------------------------------------------------------------- /cls/models/mobilenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kaiyuyue/mgd/HEAD/cls/models/mobilenetv2.py -------------------------------------------------------------------------------- /cls/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kaiyuyue/mgd/HEAD/cls/models/resnet.py -------------------------------------------------------------------------------- /cls/models/shufflenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kaiyuyue/mgd/HEAD/cls/models/shufflenetv2.py -------------------------------------------------------------------------------- /cls/models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kaiyuyue/mgd/HEAD/cls/models/utils.py -------------------------------------------------------------------------------- /cls/solver/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kaiyuyue/mgd/HEAD/cls/solver/__init__.py -------------------------------------------------------------------------------- /cls/solver/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kaiyuyue/mgd/HEAD/cls/solver/lr_scheduler.py -------------------------------------------------------------------------------- /d2/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kaiyuyue/mgd/HEAD/d2/README.md -------------------------------------------------------------------------------- /d2/configs/retinanet_torchvision_R_18_FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kaiyuyue/mgd/HEAD/d2/configs/retinanet_torchvision_R_18_FPN_1x.yaml -------------------------------------------------------------------------------- /d2/configs/retinanet_torchvision_R_50_FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kaiyuyue/mgd/HEAD/d2/configs/retinanet_torchvision_R_50_FPN_1x.yaml -------------------------------------------------------------------------------- /d2/d2/engine/defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kaiyuyue/mgd/HEAD/d2/d2/engine/defaults.py -------------------------------------------------------------------------------- /d2/d2/modeling/backbone/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kaiyuyue/mgd/HEAD/d2/d2/modeling/backbone/resnet.py -------------------------------------------------------------------------------- /d2/d2/modeling/meta_arch/retinanet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kaiyuyue/mgd/HEAD/d2/d2/modeling/meta_arch/retinanet.py -------------------------------------------------------------------------------- /d2/mgd/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kaiyuyue/mgd/HEAD/d2/mgd/__init__.py -------------------------------------------------------------------------------- /d2/mgd/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kaiyuyue/mgd/HEAD/d2/mgd/builder.py -------------------------------------------------------------------------------- /d2/mgd/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kaiyuyue/mgd/HEAD/d2/mgd/config.py -------------------------------------------------------------------------------- /d2/scripts/convert-output-to-d2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kaiyuyue/mgd/HEAD/d2/scripts/convert-output-to-d2.py -------------------------------------------------------------------------------- /d2/scripts/merge-models-into-one.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kaiyuyue/mgd/HEAD/d2/scripts/merge-models-into-one.py -------------------------------------------------------------------------------- /d2/train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kaiyuyue/mgd/HEAD/d2/train_net.py -------------------------------------------------------------------------------- /mgd/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kaiyuyue/mgd/HEAD/mgd/__init__.py -------------------------------------------------------------------------------- /mgd/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kaiyuyue/mgd/HEAD/mgd/builder.py -------------------------------------------------------------------------------- /mgd/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kaiyuyue/mgd/HEAD/mgd/sampler.py -------------------------------------------------------------------------------- /unsup/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kaiyuyue/mgd/HEAD/unsup/README.md -------------------------------------------------------------------------------- /unsup/main_moco_mgd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kaiyuyue/mgd/HEAD/unsup/main_moco_mgd.py -------------------------------------------------------------------------------- /unsup/mgd/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /unsup/mgd/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kaiyuyue/mgd/HEAD/unsup/mgd/builder.py -------------------------------------------------------------------------------- /unsup/moco/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved 2 | -------------------------------------------------------------------------------- /unsup/moco/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kaiyuyue/mgd/HEAD/unsup/moco/builder.py -------------------------------------------------------------------------------- /unsup/moco/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kaiyuyue/mgd/HEAD/unsup/moco/loader.py -------------------------------------------------------------------------------- /unsup/models/__init__.py: -------------------------------------------------------------------------------- 1 | from .resnet import * 2 | -------------------------------------------------------------------------------- /unsup/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/kaiyuyue/mgd/HEAD/unsup/models/resnet.py --------------------------------------------------------------------------------