├── LICENSE ├── README.md ├── architecture.png ├── cls ├── README.md ├── configs │ ├── _base_ │ │ └── models │ │ │ └── mobilenet_v1_1x.py │ ├── distillers │ │ └── mgd │ │ │ ├── res34_distill_res18_img.py │ │ │ └── res50_distill_mv1_img.py │ └── mobilenet_v1 │ │ └── mobilenet_v1.py ├── mmcls │ ├── apis │ │ └── train.py │ ├── distillation │ │ ├── __init__.py │ │ ├── builder.py │ │ ├── distillers │ │ │ ├── __init__.py │ │ │ └── classification_distiller.py │ │ └── losses │ │ │ ├── __init__.py │ │ │ └── mgd.py │ └── models │ │ └── backbones │ │ ├── __init__.py │ │ └── mobilenet_v1.py └── tools │ └── train.py ├── det ├── README.md ├── configs │ └── distillers │ │ └── mgd │ │ ├── cascade_mask_rcnn_rx101_32x4d_distill_faster_rcnn_r50_fpn_2x_coco.py │ │ ├── cascade_mask_rcnn_rx101_32x4d_distill_mask_rcnn_r50_fpn_2x_coco.py │ │ ├── reppoints_rx101_64x4d_distill_reppoints_r50_fpn_2x_coco.py │ │ ├── retina_rx101_64x4d_distill_retina_r50_fpn_2x_coco.py │ │ └── solo_r101_ms_distill_solo_r50_coco.py ├── mmdet │ ├── apis │ │ └── train.py │ └── distillation │ │ ├── __init__.py │ │ ├── builder.py │ │ ├── distillers │ │ ├── __init__.py │ │ └── detection_distiller.py │ │ └── losses │ │ ├── __init__.py │ │ └── mgd.py └── tools │ └── train.py ├── pth_transfer.py └── seg ├── README.md ├── configs ├── _base_ │ └── datasets │ │ └── cityscapes_512x512.py ├── deeplabv3 │ └── deeplabv3_r18-d8_512x512_40k_cityscapes.py ├── distillers │ └── mgd │ │ ├── psp_r101_distill_deepv3_r18_40k_512x512_city.py │ │ └── psp_r101_distill_psp_r18_40k_512x512_city.py └── pspnet │ └── pspnet_r18-d8_512x512_40k_cityscapes.py ├── mmseg ├── apis │ └── train.py └── distillation │ ├── __init__.py │ ├── builder.py │ ├── distillers │ ├── __init__.py │ └── segmentation_distiller.py │ └── losses │ ├── __init__.py │ └── mgd.py └── tools └── train.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/README.md -------------------------------------------------------------------------------- /architecture.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/architecture.png -------------------------------------------------------------------------------- /cls/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/cls/README.md -------------------------------------------------------------------------------- /cls/configs/_base_/models/mobilenet_v1_1x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/cls/configs/_base_/models/mobilenet_v1_1x.py -------------------------------------------------------------------------------- /cls/configs/distillers/mgd/res34_distill_res18_img.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/cls/configs/distillers/mgd/res34_distill_res18_img.py -------------------------------------------------------------------------------- /cls/configs/distillers/mgd/res50_distill_mv1_img.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/cls/configs/distillers/mgd/res50_distill_mv1_img.py -------------------------------------------------------------------------------- /cls/configs/mobilenet_v1/mobilenet_v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/cls/configs/mobilenet_v1/mobilenet_v1.py -------------------------------------------------------------------------------- /cls/mmcls/apis/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/cls/mmcls/apis/train.py -------------------------------------------------------------------------------- /cls/mmcls/distillation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/cls/mmcls/distillation/__init__.py -------------------------------------------------------------------------------- /cls/mmcls/distillation/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/cls/mmcls/distillation/builder.py -------------------------------------------------------------------------------- /cls/mmcls/distillation/distillers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/cls/mmcls/distillation/distillers/__init__.py -------------------------------------------------------------------------------- /cls/mmcls/distillation/distillers/classification_distiller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/cls/mmcls/distillation/distillers/classification_distiller.py -------------------------------------------------------------------------------- /cls/mmcls/distillation/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/cls/mmcls/distillation/losses/__init__.py -------------------------------------------------------------------------------- /cls/mmcls/distillation/losses/mgd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/cls/mmcls/distillation/losses/mgd.py -------------------------------------------------------------------------------- /cls/mmcls/models/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/cls/mmcls/models/backbones/__init__.py -------------------------------------------------------------------------------- /cls/mmcls/models/backbones/mobilenet_v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/cls/mmcls/models/backbones/mobilenet_v1.py -------------------------------------------------------------------------------- /cls/tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/cls/tools/train.py -------------------------------------------------------------------------------- /det/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/det/README.md -------------------------------------------------------------------------------- /det/configs/distillers/mgd/cascade_mask_rcnn_rx101_32x4d_distill_faster_rcnn_r50_fpn_2x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/det/configs/distillers/mgd/cascade_mask_rcnn_rx101_32x4d_distill_faster_rcnn_r50_fpn_2x_coco.py -------------------------------------------------------------------------------- /det/configs/distillers/mgd/cascade_mask_rcnn_rx101_32x4d_distill_mask_rcnn_r50_fpn_2x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/det/configs/distillers/mgd/cascade_mask_rcnn_rx101_32x4d_distill_mask_rcnn_r50_fpn_2x_coco.py -------------------------------------------------------------------------------- /det/configs/distillers/mgd/reppoints_rx101_64x4d_distill_reppoints_r50_fpn_2x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/det/configs/distillers/mgd/reppoints_rx101_64x4d_distill_reppoints_r50_fpn_2x_coco.py -------------------------------------------------------------------------------- /det/configs/distillers/mgd/retina_rx101_64x4d_distill_retina_r50_fpn_2x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/det/configs/distillers/mgd/retina_rx101_64x4d_distill_retina_r50_fpn_2x_coco.py -------------------------------------------------------------------------------- /det/configs/distillers/mgd/solo_r101_ms_distill_solo_r50_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/det/configs/distillers/mgd/solo_r101_ms_distill_solo_r50_coco.py -------------------------------------------------------------------------------- /det/mmdet/apis/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/det/mmdet/apis/train.py -------------------------------------------------------------------------------- /det/mmdet/distillation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/det/mmdet/distillation/__init__.py -------------------------------------------------------------------------------- /det/mmdet/distillation/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/det/mmdet/distillation/builder.py -------------------------------------------------------------------------------- /det/mmdet/distillation/distillers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/det/mmdet/distillation/distillers/__init__.py -------------------------------------------------------------------------------- /det/mmdet/distillation/distillers/detection_distiller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/det/mmdet/distillation/distillers/detection_distiller.py -------------------------------------------------------------------------------- /det/mmdet/distillation/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/det/mmdet/distillation/losses/__init__.py -------------------------------------------------------------------------------- /det/mmdet/distillation/losses/mgd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/det/mmdet/distillation/losses/mgd.py -------------------------------------------------------------------------------- /det/tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/det/tools/train.py -------------------------------------------------------------------------------- /pth_transfer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/pth_transfer.py -------------------------------------------------------------------------------- /seg/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/seg/README.md -------------------------------------------------------------------------------- /seg/configs/_base_/datasets/cityscapes_512x512.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/seg/configs/_base_/datasets/cityscapes_512x512.py -------------------------------------------------------------------------------- /seg/configs/deeplabv3/deeplabv3_r18-d8_512x512_40k_cityscapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/seg/configs/deeplabv3/deeplabv3_r18-d8_512x512_40k_cityscapes.py -------------------------------------------------------------------------------- /seg/configs/distillers/mgd/psp_r101_distill_deepv3_r18_40k_512x512_city.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/seg/configs/distillers/mgd/psp_r101_distill_deepv3_r18_40k_512x512_city.py -------------------------------------------------------------------------------- /seg/configs/distillers/mgd/psp_r101_distill_psp_r18_40k_512x512_city.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/seg/configs/distillers/mgd/psp_r101_distill_psp_r18_40k_512x512_city.py -------------------------------------------------------------------------------- /seg/configs/pspnet/pspnet_r18-d8_512x512_40k_cityscapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/seg/configs/pspnet/pspnet_r18-d8_512x512_40k_cityscapes.py -------------------------------------------------------------------------------- /seg/mmseg/apis/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/seg/mmseg/apis/train.py -------------------------------------------------------------------------------- /seg/mmseg/distillation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/seg/mmseg/distillation/__init__.py -------------------------------------------------------------------------------- /seg/mmseg/distillation/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/seg/mmseg/distillation/builder.py -------------------------------------------------------------------------------- /seg/mmseg/distillation/distillers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/seg/mmseg/distillation/distillers/__init__.py -------------------------------------------------------------------------------- /seg/mmseg/distillation/distillers/segmentation_distiller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/seg/mmseg/distillation/distillers/segmentation_distiller.py -------------------------------------------------------------------------------- /seg/mmseg/distillation/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/seg/mmseg/distillation/losses/__init__.py -------------------------------------------------------------------------------- /seg/mmseg/distillation/losses/mgd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/seg/mmseg/distillation/losses/mgd.py -------------------------------------------------------------------------------- /seg/tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yzd-v/MGD/HEAD/seg/tools/train.py --------------------------------------------------------------------------------