Code for CVPR 2019 Paper "Libra R-CNN: Towards Balanced Learning for Object Detection" (#687)
* Code for components of Libra R-CNN * Configs and README for Libra R-CNN * update bfp * Update Model ZOO * add comments in non-local * fix shape * update bfp * update according to ck's comments * update des * update des * fix loss * fix according to ck's comments * fix activation in non-local * fix conv_mask in non-local * fix conv_mask in non-local * Remove outdated model urls * refactoring for bfp * change in_channels from list[int] to int * refactoring for nonlocal * udpate weight init of nonlocal * minor fix * update new model urls
Showing
- MODEL_ZOO.md 4 additions, 4 deletionsMODEL_ZOO.md
- README.md 1 addition, 0 deletionsREADME.md
- configs/libra_rcnn/README.md 26 additions, 0 deletionsconfigs/libra_rcnn/README.md
- configs/libra_rcnn/libra_fast_rcnn_r50_fpn_1x.py 144 additions, 0 deletionsconfigs/libra_rcnn/libra_fast_rcnn_r50_fpn_1x.py
- configs/libra_rcnn/libra_faster_rcnn_r101_fpn_1x.py 185 additions, 0 deletionsconfigs/libra_rcnn/libra_faster_rcnn_r101_fpn_1x.py
- configs/libra_rcnn/libra_faster_rcnn_r50_fpn_1x.py 185 additions, 0 deletionsconfigs/libra_rcnn/libra_faster_rcnn_r50_fpn_1x.py
- configs/libra_rcnn/libra_faster_rcnn_x101_64x4d_fpn_1x.py 187 additions, 0 deletionsconfigs/libra_rcnn/libra_faster_rcnn_x101_64x4d_fpn_1x.py
- configs/libra_rcnn/libra_retinanet_r50_fpn_1x.py 141 additions, 0 deletionsconfigs/libra_rcnn/libra_retinanet_r50_fpn_1x.py
- mmdet/core/bbox/samplers/iou_balanced_neg_sampler.py 97 additions, 26 deletionsmmdet/core/bbox/samplers/iou_balanced_neg_sampler.py
- mmdet/core/loss/__init__.py 4 additions, 2 deletionsmmdet/core/loss/__init__.py
- mmdet/core/loss/losses.py 50 additions, 13 deletionsmmdet/core/loss/losses.py
- mmdet/models/losses/__init__.py 5 additions, 1 deletionmmdet/models/losses/__init__.py
- mmdet/models/losses/balanced_l1_loss.py 31 additions, 0 deletionsmmdet/models/losses/balanced_l1_loss.py
- mmdet/models/necks/__init__.py 2 additions, 1 deletionmmdet/models/necks/__init__.py
- mmdet/models/necks/bfp.py 102 additions, 0 deletionsmmdet/models/necks/bfp.py
- mmdet/models/plugins/__init__.py 3 additions, 0 deletionsmmdet/models/plugins/__init__.py
- mmdet/models/plugins/non_local.py 114 additions, 0 deletionsmmdet/models/plugins/non_local.py
Loading
Please register or sign in to comment