panoptic-fpn_r50_fpn_ms-3x_coco.py 939 B

1234567891011121314151617181920212223242526272829303132333435
  1. _base_ = './panoptic-fpn_r50_fpn_1x_coco.py'
  2. # In mstrain 3x config, img_scale=[(1333, 640), (1333, 800)],
  3. # multiscale_mode='range'
  4. train_pipeline = [
  5. dict(type='LoadImageFromFile'),
  6. dict(
  7. type='LoadPanopticAnnotations',
  8. with_bbox=True,
  9. with_mask=True,
  10. with_seg=True),
  11. dict(
  12. type='RandomResize', scale=[(1333, 640), (1333, 800)],
  13. keep_ratio=True),
  14. dict(type='RandomFlip', prob=0.5),
  15. dict(type='PackDetInputs')
  16. ]
  17. train_dataloader = dict(dataset=dict(pipeline=train_pipeline))
  18. # TODO: Use RepeatDataset to speed up training
  19. # training schedule for 3x
  20. train_cfg = dict(max_epochs=36, val_interval=3)
  21. # learning rate
  22. param_scheduler = [
  23. dict(
  24. type='LinearLR', start_factor=0.001, by_epoch=False, begin=0, end=500),
  25. dict(
  26. type='MultiStepLR',
  27. begin=0,
  28. end=36,
  29. by_epoch=True,
  30. milestones=[24, 33],
  31. gamma=0.1)
  32. ]