libra-retinanet_r50_fpn_1x_coco.py 674 B

1234567891011121314151617181920212223242526
  1. _base_ = '../retinanet/retinanet_r50_fpn_1x_coco.py'
  2. # model settings
  3. model = dict(
  4. neck=[
  5. dict(
  6. type='FPN',
  7. in_channels=[256, 512, 1024, 2048],
  8. out_channels=256,
  9. start_level=1,
  10. add_extra_convs='on_input',
  11. num_outs=5),
  12. dict(
  13. type='BFP',
  14. in_channels=256,
  15. num_levels=5,
  16. refine_level=1,
  17. refine_type='non_local')
  18. ],
  19. bbox_head=dict(
  20. loss_bbox=dict(
  21. _delete_=True,
  22. type='BalancedL1Loss',
  23. alpha=0.5,
  24. gamma=1.5,
  25. beta=0.11,
  26. loss_weight=1.0)))