mask-rcnn_r50-contrib_fpn_gn-all_2x_coco.py 863 B

12345678910111213141516171819202122232425262728293031
  1. _base_ = '../mask_rcnn/mask-rcnn_r50_fpn_1x_coco.py'
  2. norm_cfg = dict(type='GN', num_groups=32, requires_grad=True)
  3. model = dict(
  4. backbone=dict(
  5. norm_cfg=norm_cfg,
  6. init_cfg=dict(
  7. type='Pretrained', checkpoint='open-mmlab://contrib/resnet50_gn')),
  8. neck=dict(norm_cfg=norm_cfg),
  9. roi_head=dict(
  10. bbox_head=dict(
  11. type='Shared4Conv1FCBBoxHead',
  12. conv_out_channels=256,
  13. norm_cfg=norm_cfg),
  14. mask_head=dict(norm_cfg=norm_cfg)))
  15. # learning policy
  16. max_epochs = 24
  17. train_cfg = dict(max_epochs=max_epochs)
  18. # learning rate
  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. ]