retinanet_r50_fpn_ghm-1x_coco.py 509 B

123456789101112131415161718
  1. _base_ = '../retinanet/retinanet_r50_fpn_1x_coco.py'
  2. model = dict(
  3. bbox_head=dict(
  4. loss_cls=dict(
  5. _delete_=True,
  6. type='GHMC',
  7. bins=30,
  8. momentum=0.75,
  9. use_sigmoid=True,
  10. loss_weight=1.0),
  11. loss_bbox=dict(
  12. _delete_=True,
  13. type='GHMR',
  14. mu=0.02,
  15. bins=10,
  16. momentum=0.7,
  17. loss_weight=10.0)))
  18. optim_wrapper = dict(clip_grad=dict(max_norm=35, norm_type=2))