mask-rcnn_x101-32x4d_fpn_pisa_1x_coco.py 929 B

123456789101112131415161718192021222324252627282930
  1. _base_ = '../mask_rcnn/mask-rcnn_x101-32x4d_fpn_1x_coco.py'
  2. model = dict(
  3. roi_head=dict(
  4. type='PISARoIHead',
  5. bbox_head=dict(
  6. loss_bbox=dict(type='SmoothL1Loss', beta=1.0, loss_weight=1.0))),
  7. train_cfg=dict(
  8. rpn_proposal=dict(
  9. nms_pre=2000,
  10. max_per_img=2000,
  11. nms=dict(type='nms', iou_threshold=0.7),
  12. min_bbox_size=0),
  13. rcnn=dict(
  14. sampler=dict(
  15. type='ScoreHLRSampler',
  16. num=512,
  17. pos_fraction=0.25,
  18. neg_pos_ub=-1,
  19. add_gt_as_proposals=True,
  20. k=0.5,
  21. bias=0.),
  22. isr=dict(k=2, bias=0),
  23. carl=dict(k=1, bias=0.2))),
  24. test_cfg=dict(
  25. rpn=dict(
  26. nms_pre=2000,
  27. max_per_img=2000,
  28. nms=dict(type='nms', iou_threshold=0.7),
  29. min_bbox_size=0)))