Code for "Gradient Harmonized Single-stage Detector" (#706)
* finish ghm in newest verion with AP 37.0 * add ghm config file * reformat for PEP8 * reformat for flake8 * add documents for GHM and tensorize the params * improve the docs * add readme and update configs
Showing
- configs/ghm/README.md 16 additions, 0 deletionsconfigs/ghm/README.md
- configs/ghm/retinanet_ghm_r50_fpn_1x.py 129 additions, 0 deletionsconfigs/ghm/retinanet_ghm_r50_fpn_1x.py
- mmdet/models/anchor_heads/anchor_head.py 1 addition, 1 deletionmmdet/models/anchor_heads/anchor_head.py
- mmdet/models/losses/__init__.py 2 additions, 1 deletionmmdet/models/losses/__init__.py
- mmdet/models/losses/ghm_loss.py 172 additions, 0 deletionsmmdet/models/losses/ghm_loss.py
Loading
Please register or sign in to comment