[Feature]: Add Localization Distillation for Object Detection (#4758)
* add Localization Distillation for Object Detection https://arxiv.org/abs/2102.12252 * fix lint * fix lint * fix lint * fix lint * fix lint * fix config * add kd detector * edit loss name * overide setattr but failed * move init_detector into init * seperate ld and gfocal, fix unused param error * small fix * small fix * add test, reload magic function, create kd_loss.py * del teacher model warnings * fix reference * add ignored bbox test fix docstring * small fix * docstring fix * change names * fix * fix * fix test * fix import * fix * docstring fix * fix * retest * add test * docstring fix
Showing
- configs/ld/ld_r101_gflv1_r101dcn_fpn_coco_2x.py 43 additions, 0 deletionsconfigs/ld/ld_r101_gflv1_r101dcn_fpn_coco_2x.py
- configs/ld/ld_r18_gflv1_r101_fpn_coco_1x.py 62 additions, 0 deletionsconfigs/ld/ld_r18_gflv1_r101_fpn_coco_1x.py
- configs/ld/ld_r34_gflv1_r101_fpn_coco_1x.py 19 additions, 0 deletionsconfigs/ld/ld_r34_gflv1_r101_fpn_coco_1x.py
- configs/ld/ld_r50_gflv1_r101_fpn_coco_1x.py 19 additions, 0 deletionsconfigs/ld/ld_r50_gflv1_r101_fpn_coco_1x.py
- configs/ld/readme.md 31 additions, 0 deletionsconfigs/ld/readme.md
- mmdet/models/dense_heads/__init__.py 2 additions, 1 deletionmmdet/models/dense_heads/__init__.py
- mmdet/models/dense_heads/ld_head.py 261 additions, 0 deletionsmmdet/models/dense_heads/ld_head.py
- mmdet/models/detectors/__init__.py 3 additions, 1 deletionmmdet/models/detectors/__init__.py
- mmdet/models/detectors/kd_one_stage.py 100 additions, 0 deletionsmmdet/models/detectors/kd_one_stage.py
- mmdet/models/losses/__init__.py 2 additions, 1 deletionmmdet/models/losses/__init__.py
- mmdet/models/losses/kd_loss.py 87 additions, 0 deletionsmmdet/models/losses/kd_loss.py
- tests/test_metrics/test_losses.py 31 additions, 0 deletionstests/test_metrics/test_losses.py
- tests/test_models/test_dense_heads/test_ld_head.py 120 additions, 0 deletionstests/test_models/test_dense_heads/test_ld_head.py
- tests/test_models/test_forward.py 55 additions, 0 deletionstests/test_models/test_forward.py
Loading
Please register or sign in to comment