ms-rcnn_r50_fpn_1x_coco.py 509 B

12345678910111213141516
  1. _base_ = '../mask_rcnn/mask-rcnn_r50_fpn_1x_coco.py'
  2. model = dict(
  3. type='MaskScoringRCNN',
  4. roi_head=dict(
  5. type='MaskScoringRoIHead',
  6. mask_iou_head=dict(
  7. type='MaskIoUHead',
  8. num_convs=4,
  9. num_fcs=2,
  10. roi_feat_size=14,
  11. in_channels=256,
  12. conv_out_channels=256,
  13. fc_out_channels=1024,
  14. num_classes=80)),
  15. # model training and testing settings
  16. train_cfg=dict(rcnn=dict(mask_thr_binary=0.5)))