mask_rcnn_r50_fpn_gn_2x_coco.yml 1.1 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061
  1. _BASE_: [
  2. '../datasets/coco_instance.yml',
  3. '../runtime.yml',
  4. '../mask_rcnn/_base_/optimizer_1x.yml',
  5. '../mask_rcnn/_base_/mask_rcnn_r50_fpn.yml',
  6. '../mask_rcnn/_base_/mask_fpn_reader.yml',
  7. ]
  8. weights: output/mask_rcnn_r50_fpn_gn_2x_coco/model_final
  9. MaskRCNN:
  10. backbone: ResNet
  11. neck: FPN
  12. rpn_head: RPNHead
  13. bbox_head: BBoxHead
  14. mask_head: MaskHead
  15. # post process
  16. bbox_post_process: BBoxPostProcess
  17. mask_post_process: MaskPostProcess
  18. FPN:
  19. out_channel: 256
  20. norm_type: gn
  21. BBoxHead:
  22. head: XConvNormHead
  23. roi_extractor:
  24. resolution: 7
  25. sampling_ratio: 0
  26. aligned: True
  27. bbox_assigner: BBoxAssigner
  28. XConvNormHead:
  29. num_convs: 4
  30. out_channel: 1024
  31. norm_type: gn
  32. MaskHead:
  33. head: MaskFeat
  34. roi_extractor:
  35. resolution: 14
  36. sampling_ratio: 0
  37. aligned: True
  38. mask_assigner: MaskAssigner
  39. share_bbox_feat: False
  40. MaskFeat:
  41. num_convs: 4
  42. out_channel: 256
  43. norm_type: gn
  44. epoch: 24
  45. LearningRate:
  46. base_lr: 0.01
  47. schedulers:
  48. - !PiecewiseDecay
  49. gamma: 0.1
  50. milestones: [16, 22]
  51. - !LinearWarmup
  52. start_factor: 0.1
  53. steps: 1000