retinanet_r50_fpn_1x_coco_v1.py 617 B

1234567891011121314151617
  1. _base_ = [
  2. '../_base_/models/retinanet_r50_fpn.py',
  3. '../_base_/datasets/coco_detection.py',
  4. '../_base_/schedules/schedule_1x.py', '../_base_/default_runtime.py'
  5. ]
  6. model = dict(
  7. bbox_head=dict(
  8. type='RetinaHead',
  9. anchor_generator=dict(
  10. type='LegacyAnchorGenerator',
  11. center_offset=0.5,
  12. octave_base_scale=4,
  13. scales_per_octave=3,
  14. ratios=[0.5, 1.0, 2.0],
  15. strides=[8, 16, 32, 64, 128]),
  16. bbox_coder=dict(type='LegacyDeltaXYWHBBoxCoder'),
  17. loss_bbox=dict(type='SmoothL1Loss', beta=0.11, loss_weight=1.0)))