GFocal (#3097)
* add gfocal losses * add gfl head, its related code, configs * fix 2x config bug * add most model and log in gfl README * update gfl model and log * Update transforms.py modify variables L, T, R, B to left, top, right, bottom * Update gfocal_loss.py modify variables into meaningful names: * pt -> scale_factor * a -> pos b -> pos_label * disl -> dis_left disr -> dis_right * wl -> weight_left wr -> weight_right * Update gfocal_loss.py update formats * remove duplicated get_bboxes/IoU calculation, make QFL consistent * add all docstring in gfl_head.py and gfocal_loss.py
Showing
- configs/gfl/README.md 32 additions, 0 deletionsconfigs/gfl/README.md
- configs/gfl/gfl_r101_fpn_dconv_c3-c5_mstrain_2x_coco.py 14 additions, 0 deletionsconfigs/gfl/gfl_r101_fpn_dconv_c3-c5_mstrain_2x_coco.py
- configs/gfl/gfl_r101_fpn_mstrain_2x_coco.py 12 additions, 0 deletionsconfigs/gfl/gfl_r101_fpn_mstrain_2x_coco.py
- configs/gfl/gfl_r50_fpn_1x_coco.py 57 additions, 0 deletionsconfigs/gfl/gfl_r50_fpn_1x_coco.py
- configs/gfl/gfl_r50_fpn_mstrain_2x_coco.py 22 additions, 0 deletionsconfigs/gfl/gfl_r50_fpn_mstrain_2x_coco.py
- configs/gfl/gfl_x101_32x4d_fpn_dconv_c4-c5_mstrain_2x_coco.py 17 additions, 0 deletions...igs/gfl/gfl_x101_32x4d_fpn_dconv_c4-c5_mstrain_2x_coco.py
- configs/gfl/gfl_x101_32x4d_fpn_mstrain_2x_coco.py 15 additions, 0 deletionsconfigs/gfl/gfl_x101_32x4d_fpn_mstrain_2x_coco.py
- mmdet/core/bbox/__init__.py 6 additions, 4 deletionsmmdet/core/bbox/__init__.py
- mmdet/core/bbox/transforms.py 24 additions, 0 deletionsmmdet/core/bbox/transforms.py
- mmdet/models/dense_heads/__init__.py 3 additions, 1 deletionmmdet/models/dense_heads/__init__.py
- mmdet/models/dense_heads/anchor_head.py 3 additions, 1 deletionmmdet/models/dense_heads/anchor_head.py
- mmdet/models/dense_heads/gfl_head.py 632 additions, 0 deletionsmmdet/models/dense_heads/gfl_head.py
- mmdet/models/detectors/__init__.py 2 additions, 1 deletionmmdet/models/detectors/__init__.py
- mmdet/models/detectors/gfl.py 16 additions, 0 deletionsmmdet/models/detectors/gfl.py
- mmdet/models/losses/__init__.py 3 additions, 1 deletionmmdet/models/losses/__init__.py
- mmdet/models/losses/gfocal_loss.py 189 additions, 0 deletionsmmdet/models/losses/gfocal_loss.py
Loading
Please register or sign in to comment