Refactor hungarian assigner (#4259)
* add focal_loss in hungarian_assigner.py
* update hungarian_assigner.py
* update detr config
* update doc and codes
* fix unitest
* fix unitest
* modify code format
* update docstring
* fix unitest
* fix name
* fix docformat
* fix docformat
* fix format
Co-authored-by:
fangkairen <fangkairen@sensetime.com>
Showing
- configs/detr/detr_r50_8x2_150e_coco.py 4 additions, 2 deletionsconfigs/detr/detr_r50_8x2_150e_coco.py
- mmdet/core/bbox/assigners/hungarian_assigner.py 16 additions, 36 deletionsmmdet/core/bbox/assigners/hungarian_assigner.py
- mmdet/core/bbox/match_costs/__init__.py 7 additions, 0 deletionsmmdet/core/bbox/match_costs/__init__.py
- mmdet/core/bbox/match_costs/builder.py 8 additions, 0 deletionsmmdet/core/bbox/match_costs/builder.py
- mmdet/core/bbox/match_costs/match_cost.py 184 additions, 0 deletionsmmdet/core/bbox/match_costs/match_cost.py
- mmdet/models/dense_heads/transformer_head.py 9 additions, 10 deletionsmmdet/models/dense_heads/transformer_head.py
- tests/test_assigner.py 15 additions, 3 deletionstests/test_assigner.py
- tests/test_models/test_heads.py 3 additions, 5 deletionstests/test_models/test_heads.py
Loading
Please register or sign in to comment