implus / GFocal

Generalized Focal Loss: Learning Qualified and Distributed Bounding Boxes for Dense Object Detection, NeurIPS2020
Apache License 2.0
580 stars 73 forks source link

如何快速使用GFL #2

Closed zjp99 closed 4 years ago

zjp99 commented 4 years ago

我目前使用的是focal loss,如何更改能直接使用您提出的GFL,来进行实验?

implus commented 4 years ago

Please refer to the related code in:

https://github.com/implus/GFocal/blob/master/mmdet/models/losses/gfocal_loss.py https://github.com/implus/GFocal/blob/master/mmdet/models/anchor_heads/gfl_head.py https://github.com/implus/GFocal/blob/master/configs/gfl_r50_1x.py

Thanks!