├── .gitignore ├── LICENSE ├── README.md ├── assests └── mask.png ├── mmdetection ├── README.md ├── configs │ ├── _base_ │ │ ├── datasets │ │ │ ├── cityscapes_detection.py │ │ │ ├── cityscapes_instance.py │ │ │ ├── coco_detection.py │ │ │ ├── coco_instance.py │ │ │ ├── coco_instance_semantic.py │ │ │ ├── deepfashion.py │ │ │ ├── lvis_v0.5_instance.py │ │ │ ├── lvis_v1_instance.py │ │ │ ├── voc0712.py │ │ │ └── wider_face.py │ │ ├── default_runtime.py │ │ ├── models │ │ │ ├── cascade_mask_rcnn_r50_fpn.py │ │ │ ├── cascade_rcnn_r50_fpn.py │ │ │ ├── fast_rcnn_r50_fpn.py │ │ │ ├── faster_rcnn_r50_caffe_c4.py │ │ │ ├── faster_rcnn_r50_caffe_dc5.py │ │ │ ├── faster_rcnn_r50_fpn.py │ │ │ ├── mask_rcnn_r50_caffe_c4.py │ │ │ ├── mask_rcnn_r50_fpn.py │ │ │ ├── retinanet_r50_fpn.py │ │ │ ├── rpn_r50_caffe_c4.py │ │ │ ├── rpn_r50_fpn.py │ │ │ └── ssd300.py │ │ └── schedules │ │ │ ├── schedule_1x.py │ │ │ ├── schedule_20e.py │ │ │ └── schedule_2x.py │ ├── cascade_rcnn │ │ ├── cascade_mask_rcnn_r50_fpn_20e_coco.py │ │ └── cascade_mask_rcnn_x101_64x4d_fpn_20e_coco.py │ ├── faster_rcnn │ │ ├── faster_rcnn_r101_fpn_2x_coco.py │ │ └── faster_rcnn_r50_fpn_2x_coco.py │ ├── fcos │ │ ├── fcos_r101_caffe_fpn_gn-head_mstrain_640-800_2x_coco.py │ │ └── fcos_r50_caffe_fpn_gn-head_1x_coco.py │ ├── maskd │ │ ├── cascade_mask_rcnn_x101_64x4d_fpn_20e_coco.py │ │ ├── faster_rcnn_r101_fpn_2x_coco.py │ │ ├── fcos_r101_caffe_fpn_gn-head_mstrain_640-800_2x_coco.py │ │ ├── reppoints_moment_x101_fpn_dconv_c3-c5_gn-neck+head_2x_coco.py │ │ ├── retinanet_r101_fpn_2x_coco.py │ │ └── retinanet_x101_64x4d_fpn_2x_coco.py │ ├── reppoints │ │ ├── reppoints_moment_r50_fpn_1x_coco.py │ │ ├── reppoints_moment_r50_fpn_gn-neck+head_1x_coco.py │ │ ├── reppoints_moment_r50_fpn_gn-neck+head_2x_coco.py │ │ └── reppoints_moment_x101_fpn_dconv_c3-c5_gn-neck+head_2x_coco.py │ └── retinanet │ │ ├── retinanet_r101_fpn_2x_coco.py │ │ ├── retinanet_r50_fpn_1x_coco.py │ │ ├── retinanet_r50_fpn_2x_coco.py │ │ └── retinanet_x101_64x4d_fpn_2x_coco.py ├── maskd.py ├── maskd_hook.py └── tools │ ├── dist_test.sh │ ├── dist_train.sh │ ├── slurm_test.sh │ ├── slurm_train.sh │ ├── test.py │ └── train.py ├── mmrazor ├── .github │ ├── CONTRIBUTING.md │ ├── ISSUE_TEMPLATE │ │ ├── ----.md │ │ ├── ---.md │ │ ├── ---bug.md │ │ ├── bug_report.md │ │ ├── config.yml │ │ ├── feature_request.md │ │ └── general-questions.md │ ├── pull_request_template.md │ └── workflows │ │ ├── build.yml │ │ ├── deploy.yml │ │ └── publish-to-pypi.yml ├── .gitignore ├── .pre-commit-config.yaml ├── .readthedocs.yml ├── LICENSE ├── README.md ├── configs │ ├── _base_ │ │ ├── datasets │ │ │ ├── mmcls │ │ │ │ ├── cifar100_bs16.py │ │ │ │ ├── cifar10_bs16.py │ │ │ │ ├── cifar10_bs96_cutout.py │ │ │ │ ├── imagenet_bs128_colorjittor.py │ │ │ │ ├── imagenet_bs256_autoslim.py │ │ │ │ ├── imagenet_bs32.py │ │ │ │ ├── imagenet_bs32_pil_resize.py │ │ │ │ ├── imagenet_bs64.py │ │ │ │ ├── imagenet_bs64_autoaug.py │ │ │ │ ├── imagenet_bs64_pil_resize.py │ │ │ │ ├── imagenet_bs64_pil_resize_autoaug.py │ │ │ │ ├── imagenet_bs64_swin_224.py │ │ │ │ └── imagenet_bs64_swin_384.py │ │ │ ├── mmdet │ │ │ │ ├── cityscapes_detection.py │ │ │ │ ├── cityscapes_instance.py │ │ │ │ ├── coco_detection.py │ │ │ │ ├── coco_instance.py │ │ │ │ ├── coco_instance_semantic.py │ │ │ │ ├── coco_panoptic.py │ │ │ │ ├── deepfashion.py │ │ │ │ ├── lvis_v0.5_instance.py │ │ │ │ ├── lvis_v1_instance.py │ │ │ │ ├── voc0712.py │ │ │ │ └── wider_face.py │ │ │ └── mmseg │ │ │ │ ├── ade20k.py │ │ │ │ ├── chase_db1.py │ │ │ │ ├── cityscapes.py │ │ │ │ ├── cityscapes_1024x1024.py │ │ │ │ ├── cityscapes_769x769.py │ │ │ │ ├── cityscapes_832x832.py │ │ │ │ ├── coco-stuff10k.py │ │ │ │ ├── coco-stuff164k.py │ │ │ │ ├── drive.py │ │ │ │ ├── hrf.py │ │ │ │ ├── pascal_context.py │ │ │ │ ├── pascal_context_59.py │ │ │ │ ├── pascal_voc12.py │ │ │ │ ├── pascal_voc12_aug.py │ │ │ │ └── stare.py │ │ ├── mmcls_runtime.py │ │ ├── mmdet_runtime.py │ │ ├── mmseg_runtime.py │ │ └── schedules │ │ │ ├── mmcls │ │ │ ├── cifar10_bs128.py │ │ │ ├── imagenet_bs1024_adamw_swin.py │ │ │ ├── imagenet_bs1024_linearlr_bn_nowd.py │ │ │ ├── imagenet_bs1024_spos.py │ │ │ ├── imagenet_bs2048.py │ │ │ ├── imagenet_bs2048_AdamW.py │ │ │ ├── imagenet_bs2048_autoslim.py │ │ │ ├── imagenet_bs2048_coslr.py │ │ │ ├── imagenet_bs256.py │ │ │ ├── imagenet_bs256_140e.py │ │ │ ├── imagenet_bs256_200e_coslr_warmup.py │ │ │ ├── imagenet_bs256_coslr.py │ │ │ ├── imagenet_bs256_epochstep.py │ │ │ └── imagenet_bs4096_AdamW.py │ │ │ ├── mmdet │ │ │ ├── schedule_1x.py │ │ │ └── schedule_2x.py │ │ │ └── mmseg │ │ │ ├── schedule_160k.py │ │ │ ├── schedule_20k.py │ │ │ ├── schedule_320k.py │ │ │ ├── schedule_40k.py │ │ │ └── schedule_80k.py │ └── distill │ │ ├── dist │ │ ├── dist+mimic_cascade_mask_rcnn_x101-fpn_x50_coco.py │ │ ├── dist+mimic_retinanet_x101-retinanet-r50_coco.py │ │ ├── dist_cascade_mask_rcnn_x101-fpn_x50_coco.py │ │ └── dist_retinanet_x101-retinanet-r50_coco.py │ │ └── maskd │ │ ├── cascade_mask_rcnn_x101-fpn_x50_coco.py │ │ ├── fcos_r101-fcos_r50_coco.py │ │ ├── fpn_r101-fpn_r50_coco.py │ │ ├── reppoints_x101-reppoints-r50_coco.py │ │ ├── retinanet_r101-retinanet_r50_coco.py │ │ └── retinanet_x101-retinanet_r50_coco.py ├── docker │ ├── Dockerfile │ └── serve │ │ ├── Dockerfile │ │ ├── config.properties │ │ └── entrypoint.sh ├── docs │ └── en │ │ ├── Makefile │ │ ├── _static │ │ ├── css │ │ │ └── readthedocs.css │ │ └── image │ │ │ └── mmrazor-logo.png │ │ ├── api.rst │ │ ├── conf.py │ │ ├── gen_model_zoo.py │ │ ├── get_started.md │ │ ├── imgs │ │ ├── model_zoo │ │ │ ├── autoslim │ │ │ │ └── pipeline.png │ │ │ ├── crd │ │ │ │ └── pipeline.jpg │ │ │ ├── cwd │ │ │ │ └── pipeline.png │ │ │ ├── darts │ │ │ │ └── pipeline.png │ │ │ ├── detnas │ │ │ │ └── pipeline.jpg │ │ │ ├── kd │ │ │ │ └── pipeline.png │ │ │ ├── meal_v2 │ │ │ │ └── pipeline.png │ │ │ ├── spos │ │ │ │ └── pipeline.jpg │ │ │ └── wsld │ │ │ │ └── pipeline.png │ │ └── tutorials │ │ │ └── overview │ │ │ ├── design_and_implement.png │ │ │ └── overview.png │ │ ├── index.rst │ │ ├── make.bat │ │ ├── model_zoo.md │ │ ├── test.md │ │ ├── train.md │ │ └── tutorials │ │ ├── Tutorial_1_overview.md │ │ ├── Tutorial_2_learn_about_configs.md │ │ ├── Tutorial_3_customize_architectures.md │ │ ├── Tutorial_4_customize_nas_algorithms.md │ │ ├── Tutorial_5_customize_pruning_algorithms.md │ │ ├── Tutorial_6_customize_kd_algorithms.md │ │ ├── Tutorial_7_customize_mixed_algorithms_with_out_algorithms_components.md │ │ ├── Tutorial_8_apply_existing_algorithms_to_new_tasks.md │ │ └── index.rst ├── mmrazor │ ├── __init__.py │ ├── apis │ │ ├── __init__.py │ │ ├── mmcls │ │ │ ├── __init__.py │ │ │ └── train.py │ │ ├── mmdet │ │ │ ├── __init__.py │ │ │ └── train.py │ │ └── mmseg │ │ │ ├── __init__.py │ │ │ └── train.py │ ├── core │ │ ├── __init__.py │ │ ├── builder.py │ │ ├── distributed_wrapper.py │ │ ├── hooks │ │ │ ├── __init__.py │ │ │ ├── drop_path_prob.py │ │ │ ├── sampler_seed.py │ │ │ └── search_subnet.py │ │ ├── optimizer │ │ │ ├── __init__.py │ │ │ └── builder.py │ │ ├── runners │ │ │ ├── __init__.py │ │ │ ├── epoch_based_runner.py │ │ │ └── iter_based_runner.py │ │ ├── searcher │ │ │ ├── __init__.py │ │ │ ├── evolution_search.py │ │ │ └── greedy_search.py │ │ └── utils │ │ │ ├── __init__.py │ │ │ ├── broadcast.py │ │ │ ├── env_utils.py │ │ │ └── lr.py │ ├── datasets │ │ ├── __init__.py │ │ └── utils.py │ ├── models │ │ ├── __init__.py │ │ ├── algorithms │ │ │ ├── __init__.py │ │ │ ├── align_method_kd.py │ │ │ ├── autoslim.py │ │ │ ├── base.py │ │ │ ├── darts.py │ │ │ ├── detnas.py │ │ │ ├── general_distill.py │ │ │ └── spos.py │ │ ├── architectures │ │ │ ├── __init__.py │ │ │ ├── base.py │ │ │ ├── components │ │ │ │ ├── __init__.py │ │ │ │ ├── backbones │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── darts_backbone.py │ │ │ │ │ └── searchable_shufflenet_v2.py │ │ │ │ ├── heads │ │ │ │ │ ├── __init__.py │ │ │ │ │ ├── darts_head.py │ │ │ │ │ └── no_bias_fc_head.py │ │ │ │ └── necks │ │ │ │ │ └── __init__.py │ │ │ ├── mmcls.py │ │ │ ├── mmdet.py │ │ │ ├── mmseg.py │ │ │ └── utils │ │ │ │ ├── __init__.py │ │ │ │ └── placeholder.py │ │ ├── builder.py │ │ ├── distillers │ │ │ ├── __init__.py │ │ │ ├── base.py │ │ │ ├── self_distiller.py │ │ │ └── single_teacher.py │ │ ├── losses │ │ │ ├── __init__.py │ │ │ ├── cwd.py │ │ │ ├── dist_kd.py │ │ │ ├── kl_divergence.py │ │ │ ├── maskd.py │ │ │ ├── mse.py │ │ │ ├── soft_cross_entropy.py │ │ │ └── weighted_soft_label_distillation.py │ │ ├── mutables │ │ │ ├── __init__.py │ │ │ ├── mutable_edge.py │ │ │ ├── mutable_module.py │ │ │ └── mutable_op.py │ │ ├── mutators │ │ │ ├── __init__.py │ │ │ ├── base.py │ │ │ ├── darts_mutator.py │ │ │ ├── differentiable_mutator.py │ │ │ └── one_shot_mutator.py │ │ ├── ops │ │ │ ├── __init__.py │ │ │ ├── base.py │ │ │ ├── darts_series.py │ │ │ └── shufflenet_series.py │ │ ├── pruners │ │ │ ├── __init__.py │ │ │ ├── ratio_pruning.py │ │ │ ├── structure_pruning.py │ │ │ └── utils │ │ │ │ ├── __init__.py │ │ │ │ └── switchable_bn.py │ │ └── utils │ │ │ ├── __init__.py │ │ │ ├── kd_bbox_head.py │ │ │ └── misc.py │ └── version.py ├── requirements.txt ├── requirements │ ├── docs.txt │ ├── mminstall.txt │ ├── optional.txt │ ├── readthedocs.txt │ ├── runtime.txt │ └── tests.txt ├── resources │ ├── design_and_implement.png │ ├── mmrazor-logo.png │ └── zhihu_qrcode.jpg ├── setup.cfg ├── setup.py ├── tests │ └── test_models │ │ ├── test_algorithms │ │ ├── test_algorithm.py │ │ └── utils.py │ │ ├── test_architecture.py │ │ ├── test_mutable.py │ │ ├── test_mutator.py │ │ ├── test_op.py │ │ └── test_pruner.py └── tools │ ├── misc │ ├── get_flops.py │ └── print_config.py │ ├── mmcls │ ├── search_mmcls.py │ ├── slurm_search_mmcls.sh │ ├── slurm_test_mmcls.sh │ ├── slurm_train_mmcls.sh │ ├── test_mmcls.py │ └── train_mmcls.py │ ├── mmdet │ ├── search_mmdet.py │ ├── slurm_search_mmdet.sh │ ├── slurm_test_mmdet.sh │ ├── slurm_train_mmdet.sh │ ├── test_mmdet.py │ └── train_mmdet.py │ ├── mmseg │ ├── slurm_test_mmseg.sh │ ├── slurm_train_mmseg.sh │ ├── test_mmseg.py │ └── train_mmseg.py │ └── model_converters │ ├── publish_model.py │ └── split_checkpoint.py └── segmentation ├── README.md ├── dataset ├── datasets.py └── list │ └── cityscapes │ ├── test.lst │ ├── train++.lst │ ├── train+.lst │ ├── train.lst │ ├── trainval.lst │ └── val.lst ├── eval.py ├── eval_vis_mask.py ├── losses ├── loss.py ├── maskd.py └── memory_loss.py ├── models ├── __init__.py ├── base_models │ ├── __init__.py │ ├── hrnet.py │ ├── mobilenetv2.py │ ├── resnet.py │ └── resnetv1b.py ├── deeplabv3.py ├── deeplabv3_mobile.py ├── model_zoo.py ├── psp_mobile.py ├── pspnet.py └── segbase.py ├── test.py ├── train_baseline.py ├── train_mask_module.py ├── train_maskd.py ├── train_scripts ├── train_baseline │ ├── deeplabv3_mbv2.sh │ ├── deeplabv3_res101.sh │ ├── deeplabv3_res18.sh │ ├── deeplabv3_res18_unpretrained.sh │ └── pspnet_res18.sh ├── train_mask_module │ ├── deeplabv3_res101.sh │ └── vis_deeplabv3_res101.sh └── train_maskd │ ├── deeplabv3_mbv2.sh │ ├── deeplabv3_res18.sh │ ├── deeplabv3_res18_rinit.sh │ ├── pspnet_res18.sh │ ├── pspnet_res18_rinit.sh │ ├── test_deeplabv3_mbv2.sh │ ├── test_deeplabv3_res18.sh │ └── test_pspnet_res18.sh ├── utils ├── __init__.py ├── distributed.py ├── flops.py ├── logger.py ├── score.py └── visualize.py └── work_dirs └── dv3-r101 ├── deeplabv3_resnet101_citys_best_model.pth └── seg_mask_module_deeplabv3_resnet101_citys_log.txt /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/README.md -------------------------------------------------------------------------------- /assests/mask.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/assests/mask.png -------------------------------------------------------------------------------- /mmdetection/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/README.md -------------------------------------------------------------------------------- /mmdetection/configs/_base_/datasets/cityscapes_detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/_base_/datasets/cityscapes_detection.py -------------------------------------------------------------------------------- /mmdetection/configs/_base_/datasets/cityscapes_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/_base_/datasets/cityscapes_instance.py -------------------------------------------------------------------------------- /mmdetection/configs/_base_/datasets/coco_detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/_base_/datasets/coco_detection.py -------------------------------------------------------------------------------- /mmdetection/configs/_base_/datasets/coco_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/_base_/datasets/coco_instance.py -------------------------------------------------------------------------------- /mmdetection/configs/_base_/datasets/coco_instance_semantic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/_base_/datasets/coco_instance_semantic.py -------------------------------------------------------------------------------- /mmdetection/configs/_base_/datasets/deepfashion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/_base_/datasets/deepfashion.py -------------------------------------------------------------------------------- /mmdetection/configs/_base_/datasets/lvis_v0.5_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/_base_/datasets/lvis_v0.5_instance.py -------------------------------------------------------------------------------- /mmdetection/configs/_base_/datasets/lvis_v1_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/_base_/datasets/lvis_v1_instance.py -------------------------------------------------------------------------------- /mmdetection/configs/_base_/datasets/voc0712.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/_base_/datasets/voc0712.py -------------------------------------------------------------------------------- /mmdetection/configs/_base_/datasets/wider_face.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/_base_/datasets/wider_face.py -------------------------------------------------------------------------------- /mmdetection/configs/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/_base_/default_runtime.py -------------------------------------------------------------------------------- /mmdetection/configs/_base_/models/cascade_mask_rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/_base_/models/cascade_mask_rcnn_r50_fpn.py -------------------------------------------------------------------------------- /mmdetection/configs/_base_/models/cascade_rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/_base_/models/cascade_rcnn_r50_fpn.py -------------------------------------------------------------------------------- /mmdetection/configs/_base_/models/fast_rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/_base_/models/fast_rcnn_r50_fpn.py -------------------------------------------------------------------------------- /mmdetection/configs/_base_/models/faster_rcnn_r50_caffe_c4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/_base_/models/faster_rcnn_r50_caffe_c4.py -------------------------------------------------------------------------------- /mmdetection/configs/_base_/models/faster_rcnn_r50_caffe_dc5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/_base_/models/faster_rcnn_r50_caffe_dc5.py -------------------------------------------------------------------------------- /mmdetection/configs/_base_/models/faster_rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/_base_/models/faster_rcnn_r50_fpn.py -------------------------------------------------------------------------------- /mmdetection/configs/_base_/models/mask_rcnn_r50_caffe_c4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/_base_/models/mask_rcnn_r50_caffe_c4.py -------------------------------------------------------------------------------- /mmdetection/configs/_base_/models/mask_rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/_base_/models/mask_rcnn_r50_fpn.py -------------------------------------------------------------------------------- /mmdetection/configs/_base_/models/retinanet_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/_base_/models/retinanet_r50_fpn.py -------------------------------------------------------------------------------- /mmdetection/configs/_base_/models/rpn_r50_caffe_c4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/_base_/models/rpn_r50_caffe_c4.py -------------------------------------------------------------------------------- /mmdetection/configs/_base_/models/rpn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/_base_/models/rpn_r50_fpn.py -------------------------------------------------------------------------------- /mmdetection/configs/_base_/models/ssd300.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/_base_/models/ssd300.py -------------------------------------------------------------------------------- /mmdetection/configs/_base_/schedules/schedule_1x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/_base_/schedules/schedule_1x.py -------------------------------------------------------------------------------- /mmdetection/configs/_base_/schedules/schedule_20e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/_base_/schedules/schedule_20e.py -------------------------------------------------------------------------------- /mmdetection/configs/_base_/schedules/schedule_2x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/_base_/schedules/schedule_2x.py -------------------------------------------------------------------------------- /mmdetection/configs/cascade_rcnn/cascade_mask_rcnn_r50_fpn_20e_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/cascade_rcnn/cascade_mask_rcnn_r50_fpn_20e_coco.py -------------------------------------------------------------------------------- /mmdetection/configs/cascade_rcnn/cascade_mask_rcnn_x101_64x4d_fpn_20e_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/cascade_rcnn/cascade_mask_rcnn_x101_64x4d_fpn_20e_coco.py -------------------------------------------------------------------------------- /mmdetection/configs/faster_rcnn/faster_rcnn_r101_fpn_2x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/faster_rcnn/faster_rcnn_r101_fpn_2x_coco.py -------------------------------------------------------------------------------- /mmdetection/configs/faster_rcnn/faster_rcnn_r50_fpn_2x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/faster_rcnn/faster_rcnn_r50_fpn_2x_coco.py -------------------------------------------------------------------------------- /mmdetection/configs/fcos/fcos_r101_caffe_fpn_gn-head_mstrain_640-800_2x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/fcos/fcos_r101_caffe_fpn_gn-head_mstrain_640-800_2x_coco.py -------------------------------------------------------------------------------- /mmdetection/configs/fcos/fcos_r50_caffe_fpn_gn-head_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/fcos/fcos_r50_caffe_fpn_gn-head_1x_coco.py -------------------------------------------------------------------------------- /mmdetection/configs/maskd/cascade_mask_rcnn_x101_64x4d_fpn_20e_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/maskd/cascade_mask_rcnn_x101_64x4d_fpn_20e_coco.py -------------------------------------------------------------------------------- /mmdetection/configs/maskd/faster_rcnn_r101_fpn_2x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/maskd/faster_rcnn_r101_fpn_2x_coco.py -------------------------------------------------------------------------------- /mmdetection/configs/maskd/fcos_r101_caffe_fpn_gn-head_mstrain_640-800_2x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/maskd/fcos_r101_caffe_fpn_gn-head_mstrain_640-800_2x_coco.py -------------------------------------------------------------------------------- /mmdetection/configs/maskd/reppoints_moment_x101_fpn_dconv_c3-c5_gn-neck+head_2x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/maskd/reppoints_moment_x101_fpn_dconv_c3-c5_gn-neck+head_2x_coco.py -------------------------------------------------------------------------------- /mmdetection/configs/maskd/retinanet_r101_fpn_2x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/maskd/retinanet_r101_fpn_2x_coco.py -------------------------------------------------------------------------------- /mmdetection/configs/maskd/retinanet_x101_64x4d_fpn_2x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/maskd/retinanet_x101_64x4d_fpn_2x_coco.py -------------------------------------------------------------------------------- /mmdetection/configs/reppoints/reppoints_moment_r50_fpn_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/reppoints/reppoints_moment_r50_fpn_1x_coco.py -------------------------------------------------------------------------------- /mmdetection/configs/reppoints/reppoints_moment_r50_fpn_gn-neck+head_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/reppoints/reppoints_moment_r50_fpn_gn-neck+head_1x_coco.py -------------------------------------------------------------------------------- /mmdetection/configs/reppoints/reppoints_moment_r50_fpn_gn-neck+head_2x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/reppoints/reppoints_moment_r50_fpn_gn-neck+head_2x_coco.py -------------------------------------------------------------------------------- /mmdetection/configs/reppoints/reppoints_moment_x101_fpn_dconv_c3-c5_gn-neck+head_2x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/reppoints/reppoints_moment_x101_fpn_dconv_c3-c5_gn-neck+head_2x_coco.py -------------------------------------------------------------------------------- /mmdetection/configs/retinanet/retinanet_r101_fpn_2x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/retinanet/retinanet_r101_fpn_2x_coco.py -------------------------------------------------------------------------------- /mmdetection/configs/retinanet/retinanet_r50_fpn_1x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/retinanet/retinanet_r50_fpn_1x_coco.py -------------------------------------------------------------------------------- /mmdetection/configs/retinanet/retinanet_r50_fpn_2x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/retinanet/retinanet_r50_fpn_2x_coco.py -------------------------------------------------------------------------------- /mmdetection/configs/retinanet/retinanet_x101_64x4d_fpn_2x_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/configs/retinanet/retinanet_x101_64x4d_fpn_2x_coco.py -------------------------------------------------------------------------------- /mmdetection/maskd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/maskd.py -------------------------------------------------------------------------------- /mmdetection/maskd_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/maskd_hook.py -------------------------------------------------------------------------------- /mmdetection/tools/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/tools/dist_test.sh -------------------------------------------------------------------------------- /mmdetection/tools/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/tools/dist_train.sh -------------------------------------------------------------------------------- /mmdetection/tools/slurm_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/tools/slurm_test.sh -------------------------------------------------------------------------------- /mmdetection/tools/slurm_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/tools/slurm_train.sh -------------------------------------------------------------------------------- /mmdetection/tools/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/tools/test.py -------------------------------------------------------------------------------- /mmdetection/tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmdetection/tools/train.py -------------------------------------------------------------------------------- /mmrazor/.github/CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/.github/CONTRIBUTING.md -------------------------------------------------------------------------------- /mmrazor/.github/ISSUE_TEMPLATE/----.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/.github/ISSUE_TEMPLATE/----.md -------------------------------------------------------------------------------- /mmrazor/.github/ISSUE_TEMPLATE/---.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/.github/ISSUE_TEMPLATE/---.md -------------------------------------------------------------------------------- /mmrazor/.github/ISSUE_TEMPLATE/---bug.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/.github/ISSUE_TEMPLATE/---bug.md -------------------------------------------------------------------------------- /mmrazor/.github/ISSUE_TEMPLATE/bug_report.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/.github/ISSUE_TEMPLATE/bug_report.md -------------------------------------------------------------------------------- /mmrazor/.github/ISSUE_TEMPLATE/config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/.github/ISSUE_TEMPLATE/config.yml -------------------------------------------------------------------------------- /mmrazor/.github/ISSUE_TEMPLATE/feature_request.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/.github/ISSUE_TEMPLATE/feature_request.md -------------------------------------------------------------------------------- /mmrazor/.github/ISSUE_TEMPLATE/general-questions.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/.github/ISSUE_TEMPLATE/general-questions.md -------------------------------------------------------------------------------- /mmrazor/.github/pull_request_template.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/.github/pull_request_template.md -------------------------------------------------------------------------------- /mmrazor/.github/workflows/build.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/.github/workflows/build.yml -------------------------------------------------------------------------------- /mmrazor/.github/workflows/deploy.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/.github/workflows/deploy.yml -------------------------------------------------------------------------------- /mmrazor/.github/workflows/publish-to-pypi.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/.github/workflows/publish-to-pypi.yml -------------------------------------------------------------------------------- /mmrazor/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/.gitignore -------------------------------------------------------------------------------- /mmrazor/.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/.pre-commit-config.yaml -------------------------------------------------------------------------------- /mmrazor/.readthedocs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/.readthedocs.yml -------------------------------------------------------------------------------- /mmrazor/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/LICENSE -------------------------------------------------------------------------------- /mmrazor/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/README.md -------------------------------------------------------------------------------- /mmrazor/configs/_base_/datasets/mmcls/cifar100_bs16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/datasets/mmcls/cifar100_bs16.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/datasets/mmcls/cifar10_bs16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/datasets/mmcls/cifar10_bs16.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/datasets/mmcls/cifar10_bs96_cutout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/datasets/mmcls/cifar10_bs96_cutout.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/datasets/mmcls/imagenet_bs128_colorjittor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/datasets/mmcls/imagenet_bs128_colorjittor.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/datasets/mmcls/imagenet_bs256_autoslim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/datasets/mmcls/imagenet_bs256_autoslim.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/datasets/mmcls/imagenet_bs32.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/datasets/mmcls/imagenet_bs32.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/datasets/mmcls/imagenet_bs32_pil_resize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/datasets/mmcls/imagenet_bs32_pil_resize.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/datasets/mmcls/imagenet_bs64.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/datasets/mmcls/imagenet_bs64.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/datasets/mmcls/imagenet_bs64_autoaug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/datasets/mmcls/imagenet_bs64_autoaug.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/datasets/mmcls/imagenet_bs64_pil_resize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/datasets/mmcls/imagenet_bs64_pil_resize.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/datasets/mmcls/imagenet_bs64_pil_resize_autoaug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/datasets/mmcls/imagenet_bs64_pil_resize_autoaug.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/datasets/mmcls/imagenet_bs64_swin_224.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/datasets/mmcls/imagenet_bs64_swin_224.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/datasets/mmcls/imagenet_bs64_swin_384.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/datasets/mmcls/imagenet_bs64_swin_384.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/datasets/mmdet/cityscapes_detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/datasets/mmdet/cityscapes_detection.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/datasets/mmdet/cityscapes_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/datasets/mmdet/cityscapes_instance.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/datasets/mmdet/coco_detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/datasets/mmdet/coco_detection.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/datasets/mmdet/coco_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/datasets/mmdet/coco_instance.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/datasets/mmdet/coco_instance_semantic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/datasets/mmdet/coco_instance_semantic.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/datasets/mmdet/coco_panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/datasets/mmdet/coco_panoptic.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/datasets/mmdet/deepfashion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/datasets/mmdet/deepfashion.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/datasets/mmdet/lvis_v0.5_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/datasets/mmdet/lvis_v0.5_instance.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/datasets/mmdet/lvis_v1_instance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/datasets/mmdet/lvis_v1_instance.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/datasets/mmdet/voc0712.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/datasets/mmdet/voc0712.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/datasets/mmdet/wider_face.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/datasets/mmdet/wider_face.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/datasets/mmseg/ade20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/datasets/mmseg/ade20k.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/datasets/mmseg/chase_db1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/datasets/mmseg/chase_db1.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/datasets/mmseg/cityscapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/datasets/mmseg/cityscapes.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/datasets/mmseg/cityscapes_1024x1024.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/datasets/mmseg/cityscapes_1024x1024.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/datasets/mmseg/cityscapes_769x769.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/datasets/mmseg/cityscapes_769x769.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/datasets/mmseg/cityscapes_832x832.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/datasets/mmseg/cityscapes_832x832.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/datasets/mmseg/coco-stuff10k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/datasets/mmseg/coco-stuff10k.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/datasets/mmseg/coco-stuff164k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/datasets/mmseg/coco-stuff164k.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/datasets/mmseg/drive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/datasets/mmseg/drive.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/datasets/mmseg/hrf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/datasets/mmseg/hrf.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/datasets/mmseg/pascal_context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/datasets/mmseg/pascal_context.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/datasets/mmseg/pascal_context_59.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/datasets/mmseg/pascal_context_59.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/datasets/mmseg/pascal_voc12.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/datasets/mmseg/pascal_voc12.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/datasets/mmseg/pascal_voc12_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/datasets/mmseg/pascal_voc12_aug.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/datasets/mmseg/stare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/datasets/mmseg/stare.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/mmcls_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/mmcls_runtime.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/mmdet_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/mmdet_runtime.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/mmseg_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/mmseg_runtime.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/schedules/mmcls/cifar10_bs128.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/schedules/mmcls/cifar10_bs128.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/schedules/mmcls/imagenet_bs1024_adamw_swin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/schedules/mmcls/imagenet_bs1024_adamw_swin.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/schedules/mmcls/imagenet_bs1024_linearlr_bn_nowd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/schedules/mmcls/imagenet_bs1024_linearlr_bn_nowd.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/schedules/mmcls/imagenet_bs1024_spos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/schedules/mmcls/imagenet_bs1024_spos.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/schedules/mmcls/imagenet_bs2048.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/schedules/mmcls/imagenet_bs2048.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/schedules/mmcls/imagenet_bs2048_AdamW.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/schedules/mmcls/imagenet_bs2048_AdamW.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/schedules/mmcls/imagenet_bs2048_autoslim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/schedules/mmcls/imagenet_bs2048_autoslim.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/schedules/mmcls/imagenet_bs2048_coslr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/schedules/mmcls/imagenet_bs2048_coslr.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/schedules/mmcls/imagenet_bs256.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/schedules/mmcls/imagenet_bs256.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/schedules/mmcls/imagenet_bs256_140e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/schedules/mmcls/imagenet_bs256_140e.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/schedules/mmcls/imagenet_bs256_200e_coslr_warmup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/schedules/mmcls/imagenet_bs256_200e_coslr_warmup.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/schedules/mmcls/imagenet_bs256_coslr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/schedules/mmcls/imagenet_bs256_coslr.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/schedules/mmcls/imagenet_bs256_epochstep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/schedules/mmcls/imagenet_bs256_epochstep.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/schedules/mmcls/imagenet_bs4096_AdamW.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/schedules/mmcls/imagenet_bs4096_AdamW.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/schedules/mmdet/schedule_1x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/schedules/mmdet/schedule_1x.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/schedules/mmdet/schedule_2x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/schedules/mmdet/schedule_2x.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/schedules/mmseg/schedule_160k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/schedules/mmseg/schedule_160k.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/schedules/mmseg/schedule_20k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/schedules/mmseg/schedule_20k.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/schedules/mmseg/schedule_320k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/schedules/mmseg/schedule_320k.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/schedules/mmseg/schedule_40k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/schedules/mmseg/schedule_40k.py -------------------------------------------------------------------------------- /mmrazor/configs/_base_/schedules/mmseg/schedule_80k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/_base_/schedules/mmseg/schedule_80k.py -------------------------------------------------------------------------------- /mmrazor/configs/distill/dist/dist+mimic_cascade_mask_rcnn_x101-fpn_x50_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/distill/dist/dist+mimic_cascade_mask_rcnn_x101-fpn_x50_coco.py -------------------------------------------------------------------------------- /mmrazor/configs/distill/dist/dist+mimic_retinanet_x101-retinanet-r50_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/distill/dist/dist+mimic_retinanet_x101-retinanet-r50_coco.py -------------------------------------------------------------------------------- /mmrazor/configs/distill/dist/dist_cascade_mask_rcnn_x101-fpn_x50_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/distill/dist/dist_cascade_mask_rcnn_x101-fpn_x50_coco.py -------------------------------------------------------------------------------- /mmrazor/configs/distill/dist/dist_retinanet_x101-retinanet-r50_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/distill/dist/dist_retinanet_x101-retinanet-r50_coco.py -------------------------------------------------------------------------------- /mmrazor/configs/distill/maskd/cascade_mask_rcnn_x101-fpn_x50_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/distill/maskd/cascade_mask_rcnn_x101-fpn_x50_coco.py -------------------------------------------------------------------------------- /mmrazor/configs/distill/maskd/fcos_r101-fcos_r50_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/distill/maskd/fcos_r101-fcos_r50_coco.py -------------------------------------------------------------------------------- /mmrazor/configs/distill/maskd/fpn_r101-fpn_r50_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/distill/maskd/fpn_r101-fpn_r50_coco.py -------------------------------------------------------------------------------- /mmrazor/configs/distill/maskd/reppoints_x101-reppoints-r50_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/distill/maskd/reppoints_x101-reppoints-r50_coco.py -------------------------------------------------------------------------------- /mmrazor/configs/distill/maskd/retinanet_r101-retinanet_r50_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/distill/maskd/retinanet_r101-retinanet_r50_coco.py -------------------------------------------------------------------------------- /mmrazor/configs/distill/maskd/retinanet_x101-retinanet_r50_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/configs/distill/maskd/retinanet_x101-retinanet_r50_coco.py -------------------------------------------------------------------------------- /mmrazor/docker/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/docker/Dockerfile -------------------------------------------------------------------------------- /mmrazor/docker/serve/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/docker/serve/Dockerfile -------------------------------------------------------------------------------- /mmrazor/docker/serve/config.properties: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/docker/serve/config.properties -------------------------------------------------------------------------------- /mmrazor/docker/serve/entrypoint.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/docker/serve/entrypoint.sh -------------------------------------------------------------------------------- /mmrazor/docs/en/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/docs/en/Makefile -------------------------------------------------------------------------------- /mmrazor/docs/en/_static/css/readthedocs.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/docs/en/_static/css/readthedocs.css -------------------------------------------------------------------------------- /mmrazor/docs/en/_static/image/mmrazor-logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/docs/en/_static/image/mmrazor-logo.png -------------------------------------------------------------------------------- /mmrazor/docs/en/api.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/docs/en/api.rst -------------------------------------------------------------------------------- /mmrazor/docs/en/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/docs/en/conf.py -------------------------------------------------------------------------------- /mmrazor/docs/en/gen_model_zoo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/docs/en/gen_model_zoo.py -------------------------------------------------------------------------------- /mmrazor/docs/en/get_started.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/docs/en/get_started.md -------------------------------------------------------------------------------- /mmrazor/docs/en/imgs/model_zoo/autoslim/pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/docs/en/imgs/model_zoo/autoslim/pipeline.png -------------------------------------------------------------------------------- /mmrazor/docs/en/imgs/model_zoo/crd/pipeline.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/docs/en/imgs/model_zoo/crd/pipeline.jpg -------------------------------------------------------------------------------- /mmrazor/docs/en/imgs/model_zoo/cwd/pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/docs/en/imgs/model_zoo/cwd/pipeline.png -------------------------------------------------------------------------------- /mmrazor/docs/en/imgs/model_zoo/darts/pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/docs/en/imgs/model_zoo/darts/pipeline.png -------------------------------------------------------------------------------- /mmrazor/docs/en/imgs/model_zoo/detnas/pipeline.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/docs/en/imgs/model_zoo/detnas/pipeline.jpg -------------------------------------------------------------------------------- /mmrazor/docs/en/imgs/model_zoo/kd/pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/docs/en/imgs/model_zoo/kd/pipeline.png -------------------------------------------------------------------------------- /mmrazor/docs/en/imgs/model_zoo/meal_v2/pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/docs/en/imgs/model_zoo/meal_v2/pipeline.png -------------------------------------------------------------------------------- /mmrazor/docs/en/imgs/model_zoo/spos/pipeline.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/docs/en/imgs/model_zoo/spos/pipeline.jpg -------------------------------------------------------------------------------- /mmrazor/docs/en/imgs/model_zoo/wsld/pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/docs/en/imgs/model_zoo/wsld/pipeline.png -------------------------------------------------------------------------------- /mmrazor/docs/en/imgs/tutorials/overview/design_and_implement.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/docs/en/imgs/tutorials/overview/design_and_implement.png -------------------------------------------------------------------------------- /mmrazor/docs/en/imgs/tutorials/overview/overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/docs/en/imgs/tutorials/overview/overview.png -------------------------------------------------------------------------------- /mmrazor/docs/en/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/docs/en/index.rst -------------------------------------------------------------------------------- /mmrazor/docs/en/make.bat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/docs/en/make.bat -------------------------------------------------------------------------------- /mmrazor/docs/en/model_zoo.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/docs/en/model_zoo.md -------------------------------------------------------------------------------- /mmrazor/docs/en/test.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/docs/en/test.md -------------------------------------------------------------------------------- /mmrazor/docs/en/train.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/docs/en/train.md -------------------------------------------------------------------------------- /mmrazor/docs/en/tutorials/Tutorial_1_overview.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/docs/en/tutorials/Tutorial_1_overview.md -------------------------------------------------------------------------------- /mmrazor/docs/en/tutorials/Tutorial_2_learn_about_configs.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/docs/en/tutorials/Tutorial_2_learn_about_configs.md -------------------------------------------------------------------------------- /mmrazor/docs/en/tutorials/Tutorial_3_customize_architectures.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/docs/en/tutorials/Tutorial_3_customize_architectures.md -------------------------------------------------------------------------------- /mmrazor/docs/en/tutorials/Tutorial_4_customize_nas_algorithms.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/docs/en/tutorials/Tutorial_4_customize_nas_algorithms.md -------------------------------------------------------------------------------- /mmrazor/docs/en/tutorials/Tutorial_5_customize_pruning_algorithms.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/docs/en/tutorials/Tutorial_5_customize_pruning_algorithms.md -------------------------------------------------------------------------------- /mmrazor/docs/en/tutorials/Tutorial_6_customize_kd_algorithms.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/docs/en/tutorials/Tutorial_6_customize_kd_algorithms.md -------------------------------------------------------------------------------- /mmrazor/docs/en/tutorials/Tutorial_7_customize_mixed_algorithms_with_out_algorithms_components.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/docs/en/tutorials/Tutorial_7_customize_mixed_algorithms_with_out_algorithms_components.md -------------------------------------------------------------------------------- /mmrazor/docs/en/tutorials/Tutorial_8_apply_existing_algorithms_to_new_tasks.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/docs/en/tutorials/Tutorial_8_apply_existing_algorithms_to_new_tasks.md -------------------------------------------------------------------------------- /mmrazor/docs/en/tutorials/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/docs/en/tutorials/index.rst -------------------------------------------------------------------------------- /mmrazor/mmrazor/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/__init__.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/apis/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/apis/__init__.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/apis/mmcls/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/apis/mmcls/__init__.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/apis/mmcls/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/apis/mmcls/train.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/apis/mmdet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/apis/mmdet/__init__.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/apis/mmdet/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/apis/mmdet/train.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/apis/mmseg/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/apis/mmseg/__init__.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/apis/mmseg/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/apis/mmseg/train.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/core/__init__.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/core/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/core/builder.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/core/distributed_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/core/distributed_wrapper.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/core/hooks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/core/hooks/__init__.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/core/hooks/drop_path_prob.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/core/hooks/drop_path_prob.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/core/hooks/sampler_seed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/core/hooks/sampler_seed.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/core/hooks/search_subnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/core/hooks/search_subnet.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/core/optimizer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/core/optimizer/__init__.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/core/optimizer/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/core/optimizer/builder.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/core/runners/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/core/runners/__init__.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/core/runners/epoch_based_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/core/runners/epoch_based_runner.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/core/runners/iter_based_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/core/runners/iter_based_runner.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/core/searcher/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/core/searcher/__init__.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/core/searcher/evolution_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/core/searcher/evolution_search.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/core/searcher/greedy_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/core/searcher/greedy_search.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/core/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/core/utils/__init__.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/core/utils/broadcast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/core/utils/broadcast.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/core/utils/env_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/core/utils/env_utils.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/core/utils/lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/core/utils/lr.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/datasets/__init__.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/datasets/utils.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/__init__.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/algorithms/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/algorithms/__init__.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/algorithms/align_method_kd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/algorithms/align_method_kd.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/algorithms/autoslim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/algorithms/autoslim.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/algorithms/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/algorithms/base.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/algorithms/darts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/algorithms/darts.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/algorithms/detnas.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/algorithms/detnas.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/algorithms/general_distill.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/algorithms/general_distill.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/algorithms/spos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/algorithms/spos.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/architectures/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/architectures/__init__.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/architectures/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/architectures/base.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/architectures/components/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/architectures/components/__init__.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/architectures/components/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/architectures/components/backbones/__init__.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/architectures/components/backbones/darts_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/architectures/components/backbones/darts_backbone.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/architectures/components/backbones/searchable_shufflenet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/architectures/components/backbones/searchable_shufflenet_v2.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/architectures/components/heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/architectures/components/heads/__init__.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/architectures/components/heads/darts_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/architectures/components/heads/darts_head.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/architectures/components/heads/no_bias_fc_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/architectures/components/heads/no_bias_fc_head.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/architectures/components/necks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/architectures/components/necks/__init__.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/architectures/mmcls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/architectures/mmcls.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/architectures/mmdet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/architectures/mmdet.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/architectures/mmseg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/architectures/mmseg.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/architectures/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/architectures/utils/__init__.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/architectures/utils/placeholder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/architectures/utils/placeholder.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/builder.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/distillers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/distillers/__init__.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/distillers/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/distillers/base.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/distillers/self_distiller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/distillers/self_distiller.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/distillers/single_teacher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/distillers/single_teacher.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/losses/__init__.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/losses/cwd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/losses/cwd.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/losses/dist_kd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/losses/dist_kd.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/losses/kl_divergence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/losses/kl_divergence.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/losses/maskd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/losses/maskd.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/losses/mse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/losses/mse.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/losses/soft_cross_entropy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/losses/soft_cross_entropy.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/losses/weighted_soft_label_distillation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/losses/weighted_soft_label_distillation.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/mutables/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/mutables/__init__.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/mutables/mutable_edge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/mutables/mutable_edge.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/mutables/mutable_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/mutables/mutable_module.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/mutables/mutable_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/mutables/mutable_op.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/mutators/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/mutators/__init__.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/mutators/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/mutators/base.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/mutators/darts_mutator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/mutators/darts_mutator.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/mutators/differentiable_mutator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/mutators/differentiable_mutator.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/mutators/one_shot_mutator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/mutators/one_shot_mutator.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/ops/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/ops/__init__.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/ops/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/ops/base.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/ops/darts_series.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/ops/darts_series.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/ops/shufflenet_series.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/ops/shufflenet_series.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/pruners/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/pruners/__init__.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/pruners/ratio_pruning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/pruners/ratio_pruning.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/pruners/structure_pruning.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/pruners/structure_pruning.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/pruners/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/pruners/utils/__init__.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/pruners/utils/switchable_bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/pruners/utils/switchable_bn.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/utils/__init__.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/utils/kd_bbox_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/utils/kd_bbox_head.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/models/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/models/utils/misc.py -------------------------------------------------------------------------------- /mmrazor/mmrazor/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/mmrazor/version.py -------------------------------------------------------------------------------- /mmrazor/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/requirements.txt -------------------------------------------------------------------------------- /mmrazor/requirements/docs.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/requirements/docs.txt -------------------------------------------------------------------------------- /mmrazor/requirements/mminstall.txt: -------------------------------------------------------------------------------- 1 | mmcls 2 | mmcv-full>=1.3.13,<=1.5.0 3 | -------------------------------------------------------------------------------- /mmrazor/requirements/optional.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/requirements/optional.txt -------------------------------------------------------------------------------- /mmrazor/requirements/readthedocs.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/requirements/readthedocs.txt -------------------------------------------------------------------------------- /mmrazor/requirements/runtime.txt: -------------------------------------------------------------------------------- 1 | mmcls 2 | ordered_set 3 | -------------------------------------------------------------------------------- /mmrazor/requirements/tests.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/requirements/tests.txt -------------------------------------------------------------------------------- /mmrazor/resources/design_and_implement.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/resources/design_and_implement.png -------------------------------------------------------------------------------- /mmrazor/resources/mmrazor-logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/resources/mmrazor-logo.png -------------------------------------------------------------------------------- /mmrazor/resources/zhihu_qrcode.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/resources/zhihu_qrcode.jpg -------------------------------------------------------------------------------- /mmrazor/setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/setup.cfg -------------------------------------------------------------------------------- /mmrazor/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/setup.py -------------------------------------------------------------------------------- /mmrazor/tests/test_models/test_algorithms/test_algorithm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/tests/test_models/test_algorithms/test_algorithm.py -------------------------------------------------------------------------------- /mmrazor/tests/test_models/test_algorithms/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/tests/test_models/test_algorithms/utils.py -------------------------------------------------------------------------------- /mmrazor/tests/test_models/test_architecture.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/tests/test_models/test_architecture.py -------------------------------------------------------------------------------- /mmrazor/tests/test_models/test_mutable.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/tests/test_models/test_mutable.py -------------------------------------------------------------------------------- /mmrazor/tests/test_models/test_mutator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/tests/test_models/test_mutator.py -------------------------------------------------------------------------------- /mmrazor/tests/test_models/test_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/tests/test_models/test_op.py -------------------------------------------------------------------------------- /mmrazor/tests/test_models/test_pruner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/tests/test_models/test_pruner.py -------------------------------------------------------------------------------- /mmrazor/tools/misc/get_flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/tools/misc/get_flops.py -------------------------------------------------------------------------------- /mmrazor/tools/misc/print_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/tools/misc/print_config.py -------------------------------------------------------------------------------- /mmrazor/tools/mmcls/search_mmcls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/tools/mmcls/search_mmcls.py -------------------------------------------------------------------------------- /mmrazor/tools/mmcls/slurm_search_mmcls.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/tools/mmcls/slurm_search_mmcls.sh -------------------------------------------------------------------------------- /mmrazor/tools/mmcls/slurm_test_mmcls.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/tools/mmcls/slurm_test_mmcls.sh -------------------------------------------------------------------------------- /mmrazor/tools/mmcls/slurm_train_mmcls.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/tools/mmcls/slurm_train_mmcls.sh -------------------------------------------------------------------------------- /mmrazor/tools/mmcls/test_mmcls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/tools/mmcls/test_mmcls.py -------------------------------------------------------------------------------- /mmrazor/tools/mmcls/train_mmcls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/tools/mmcls/train_mmcls.py -------------------------------------------------------------------------------- /mmrazor/tools/mmdet/search_mmdet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/tools/mmdet/search_mmdet.py -------------------------------------------------------------------------------- /mmrazor/tools/mmdet/slurm_search_mmdet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/tools/mmdet/slurm_search_mmdet.sh -------------------------------------------------------------------------------- /mmrazor/tools/mmdet/slurm_test_mmdet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/tools/mmdet/slurm_test_mmdet.sh -------------------------------------------------------------------------------- /mmrazor/tools/mmdet/slurm_train_mmdet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/tools/mmdet/slurm_train_mmdet.sh -------------------------------------------------------------------------------- /mmrazor/tools/mmdet/test_mmdet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/tools/mmdet/test_mmdet.py -------------------------------------------------------------------------------- /mmrazor/tools/mmdet/train_mmdet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/tools/mmdet/train_mmdet.py -------------------------------------------------------------------------------- /mmrazor/tools/mmseg/slurm_test_mmseg.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/tools/mmseg/slurm_test_mmseg.sh -------------------------------------------------------------------------------- /mmrazor/tools/mmseg/slurm_train_mmseg.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/tools/mmseg/slurm_train_mmseg.sh -------------------------------------------------------------------------------- /mmrazor/tools/mmseg/test_mmseg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/tools/mmseg/test_mmseg.py -------------------------------------------------------------------------------- /mmrazor/tools/mmseg/train_mmseg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/tools/mmseg/train_mmseg.py -------------------------------------------------------------------------------- /mmrazor/tools/model_converters/publish_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/tools/model_converters/publish_model.py -------------------------------------------------------------------------------- /mmrazor/tools/model_converters/split_checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/mmrazor/tools/model_converters/split_checkpoint.py -------------------------------------------------------------------------------- /segmentation/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/README.md -------------------------------------------------------------------------------- /segmentation/dataset/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/dataset/datasets.py -------------------------------------------------------------------------------- /segmentation/dataset/list/cityscapes/test.lst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/dataset/list/cityscapes/test.lst -------------------------------------------------------------------------------- /segmentation/dataset/list/cityscapes/train++.lst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/dataset/list/cityscapes/train++.lst -------------------------------------------------------------------------------- /segmentation/dataset/list/cityscapes/train+.lst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/dataset/list/cityscapes/train+.lst -------------------------------------------------------------------------------- /segmentation/dataset/list/cityscapes/train.lst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/dataset/list/cityscapes/train.lst -------------------------------------------------------------------------------- /segmentation/dataset/list/cityscapes/trainval.lst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/dataset/list/cityscapes/trainval.lst -------------------------------------------------------------------------------- /segmentation/dataset/list/cityscapes/val.lst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/dataset/list/cityscapes/val.lst -------------------------------------------------------------------------------- /segmentation/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/eval.py -------------------------------------------------------------------------------- /segmentation/eval_vis_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/eval_vis_mask.py -------------------------------------------------------------------------------- /segmentation/losses/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/losses/loss.py -------------------------------------------------------------------------------- /segmentation/losses/maskd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/losses/maskd.py -------------------------------------------------------------------------------- /segmentation/losses/memory_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/losses/memory_loss.py -------------------------------------------------------------------------------- /segmentation/models/__init__.py: -------------------------------------------------------------------------------- 1 | """Model Zoo""" 2 | -------------------------------------------------------------------------------- /segmentation/models/base_models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /segmentation/models/base_models/hrnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/models/base_models/hrnet.py -------------------------------------------------------------------------------- /segmentation/models/base_models/mobilenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/models/base_models/mobilenetv2.py -------------------------------------------------------------------------------- /segmentation/models/base_models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/models/base_models/resnet.py -------------------------------------------------------------------------------- /segmentation/models/base_models/resnetv1b.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/models/base_models/resnetv1b.py -------------------------------------------------------------------------------- /segmentation/models/deeplabv3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/models/deeplabv3.py -------------------------------------------------------------------------------- /segmentation/models/deeplabv3_mobile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/models/deeplabv3_mobile.py -------------------------------------------------------------------------------- /segmentation/models/model_zoo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/models/model_zoo.py -------------------------------------------------------------------------------- /segmentation/models/psp_mobile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/models/psp_mobile.py -------------------------------------------------------------------------------- /segmentation/models/pspnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/models/pspnet.py -------------------------------------------------------------------------------- /segmentation/models/segbase.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/models/segbase.py -------------------------------------------------------------------------------- /segmentation/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/test.py -------------------------------------------------------------------------------- /segmentation/train_baseline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/train_baseline.py -------------------------------------------------------------------------------- /segmentation/train_mask_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/train_mask_module.py -------------------------------------------------------------------------------- /segmentation/train_maskd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/train_maskd.py -------------------------------------------------------------------------------- /segmentation/train_scripts/train_baseline/deeplabv3_mbv2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/train_scripts/train_baseline/deeplabv3_mbv2.sh -------------------------------------------------------------------------------- /segmentation/train_scripts/train_baseline/deeplabv3_res101.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/train_scripts/train_baseline/deeplabv3_res101.sh -------------------------------------------------------------------------------- /segmentation/train_scripts/train_baseline/deeplabv3_res18.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/train_scripts/train_baseline/deeplabv3_res18.sh -------------------------------------------------------------------------------- /segmentation/train_scripts/train_baseline/deeplabv3_res18_unpretrained.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/train_scripts/train_baseline/deeplabv3_res18_unpretrained.sh -------------------------------------------------------------------------------- /segmentation/train_scripts/train_baseline/pspnet_res18.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/train_scripts/train_baseline/pspnet_res18.sh -------------------------------------------------------------------------------- /segmentation/train_scripts/train_mask_module/deeplabv3_res101.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/train_scripts/train_mask_module/deeplabv3_res101.sh -------------------------------------------------------------------------------- /segmentation/train_scripts/train_mask_module/vis_deeplabv3_res101.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/train_scripts/train_mask_module/vis_deeplabv3_res101.sh -------------------------------------------------------------------------------- /segmentation/train_scripts/train_maskd/deeplabv3_mbv2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/train_scripts/train_maskd/deeplabv3_mbv2.sh -------------------------------------------------------------------------------- /segmentation/train_scripts/train_maskd/deeplabv3_res18.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/train_scripts/train_maskd/deeplabv3_res18.sh -------------------------------------------------------------------------------- /segmentation/train_scripts/train_maskd/deeplabv3_res18_rinit.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/train_scripts/train_maskd/deeplabv3_res18_rinit.sh -------------------------------------------------------------------------------- /segmentation/train_scripts/train_maskd/pspnet_res18.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/train_scripts/train_maskd/pspnet_res18.sh -------------------------------------------------------------------------------- /segmentation/train_scripts/train_maskd/pspnet_res18_rinit.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/train_scripts/train_maskd/pspnet_res18_rinit.sh -------------------------------------------------------------------------------- /segmentation/train_scripts/train_maskd/test_deeplabv3_mbv2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/train_scripts/train_maskd/test_deeplabv3_mbv2.sh -------------------------------------------------------------------------------- /segmentation/train_scripts/train_maskd/test_deeplabv3_res18.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/train_scripts/train_maskd/test_deeplabv3_res18.sh -------------------------------------------------------------------------------- /segmentation/train_scripts/train_maskd/test_pspnet_res18.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/train_scripts/train_maskd/test_pspnet_res18.sh -------------------------------------------------------------------------------- /segmentation/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/utils/__init__.py -------------------------------------------------------------------------------- /segmentation/utils/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/utils/distributed.py -------------------------------------------------------------------------------- /segmentation/utils/flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/utils/flops.py -------------------------------------------------------------------------------- /segmentation/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/utils/logger.py -------------------------------------------------------------------------------- /segmentation/utils/score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/utils/score.py -------------------------------------------------------------------------------- /segmentation/utils/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/utils/visualize.py -------------------------------------------------------------------------------- /segmentation/work_dirs/dv3-r101/deeplabv3_resnet101_citys_best_model.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/work_dirs/dv3-r101/deeplabv3_resnet101_citys_best_model.pth -------------------------------------------------------------------------------- /segmentation/work_dirs/dv3-r101/seg_mask_module_deeplabv3_resnet101_citys_log.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/MasKD/HEAD/segmentation/work_dirs/dv3-r101/seg_mask_module_deeplabv3_resnet101_citys_log.txt --------------------------------------------------------------------------------