mask-rcnn_r50_fpn_1x_coco_v1.py 1.2 KB

12345678910111213141516171819202122232425262728293031323334
  1. _base_ = [
  2. '../_base_/models/mask-rcnn_r50_fpn.py',
  3. '../_base_/datasets/coco_instance.py',
  4. '../_base_/schedules/schedule_1x.py', '../_base_/default_runtime.py'
  5. ]
  6. model = dict(
  7. rpn_head=dict(
  8. anchor_generator=dict(type='LegacyAnchorGenerator', center_offset=0.5),
  9. bbox_coder=dict(type='LegacyDeltaXYWHBBoxCoder'),
  10. loss_bbox=dict(type='SmoothL1Loss', beta=1.0 / 9.0, loss_weight=1.0)),
  11. roi_head=dict(
  12. bbox_roi_extractor=dict(
  13. type='SingleRoIExtractor',
  14. roi_layer=dict(
  15. type='RoIAlign',
  16. output_size=7,
  17. sampling_ratio=2,
  18. aligned=False)),
  19. mask_roi_extractor=dict(
  20. type='SingleRoIExtractor',
  21. roi_layer=dict(
  22. type='RoIAlign',
  23. output_size=14,
  24. sampling_ratio=2,
  25. aligned=False)),
  26. bbox_head=dict(
  27. bbox_coder=dict(type='LegacyDeltaXYWHBBoxCoder'),
  28. loss_bbox=dict(type='SmoothL1Loss', beta=1.0, loss_weight=1.0))),
  29. # model training and testing settings
  30. train_cfg=dict(
  31. rpn_proposal=dict(max_per_img=2000),
  32. rcnn=dict(assigner=dict(match_low_quality=True))))