queryinst_r50_fpn_ms-480-800-3x_coco.py 967 B

1234567891011121314151617181920212223242526272829303132
  1. _base_ = './queryinst_r50_fpn_1x_coco.py'
  2. train_pipeline = [
  3. dict(type='LoadImageFromFile', backend_args={{_base_.backend_args}}),
  4. dict(type='LoadAnnotations', with_bbox=True, with_mask=True),
  5. dict(
  6. type='RandomChoiceResize',
  7. scales=[(480, 1333), (512, 1333), (544, 1333), (576, 1333),
  8. (608, 1333), (640, 1333), (672, 1333), (704, 1333),
  9. (736, 1333), (768, 1333), (800, 1333)],
  10. keep_ratio=True),
  11. dict(type='RandomFlip', prob=0.5),
  12. dict(type='PackDetInputs')
  13. ]
  14. train_dataloader = dict(dataset=dict(pipeline=train_pipeline))
  15. # learning policy
  16. max_epochs = 36
  17. train_cfg = dict(type='EpochBasedTrainLoop', max_epochs=max_epochs)
  18. param_scheduler = [
  19. dict(
  20. type='LinearLR', start_factor=0.001, by_epoch=False, begin=0, end=500),
  21. dict(
  22. type='MultiStepLR',
  23. begin=0,
  24. end=max_epochs,
  25. by_epoch=True,
  26. milestones=[27, 33],
  27. gamma=0.1)
  28. ]