fovea_r50_fpn_gn-head-align_ms-640-800-4xb4-2x_coco.py 901 B

123456789101112131415161718192021222324252627282930
  1. _base_ = './fovea_r50_fpn_4xb4-1x_coco.py'
  2. model = dict(
  3. bbox_head=dict(
  4. with_deform=True,
  5. norm_cfg=dict(type='GN', num_groups=32, requires_grad=True)))
  6. train_pipeline = [
  7. dict(type='LoadImageFromFile', backend_args={{_base_.backend_args}}),
  8. dict(type='LoadAnnotations', with_bbox=True),
  9. dict(
  10. type='RandomChoiceResize',
  11. scales=[(1333, 640), (1333, 800)],
  12. keep_ratio=True),
  13. dict(type='RandomFlip', prob=0.5),
  14. dict(type='PackDetInputs')
  15. ]
  16. train_dataloader = dict(dataset=dict(pipeline=train_pipeline))
  17. # learning policy
  18. max_epochs = 24
  19. param_scheduler = [
  20. dict(
  21. type='LinearLR', start_factor=0.001, by_epoch=False, begin=0, end=500),
  22. dict(
  23. type='MultiStepLR',
  24. begin=0,
  25. end=max_epochs,
  26. by_epoch=True,
  27. milestones=[16, 22],
  28. gamma=0.1)
  29. ]
  30. train_cfg = dict(max_epochs=max_epochs)