mask-rcnn_r50-caffe_fpn_poly-1x_coco_v1.py 1019 B

12345678910111213141516171819202122232425262728293031
  1. _base_ = './mask-rcnn_r50_fpn_1x_coco.py'
  2. model = dict(
  3. # use caffe img_norm
  4. data_preprocessor=dict(
  5. mean=[103.530, 116.280, 123.675],
  6. std=[1.0, 1.0, 1.0],
  7. bgr_to_rgb=False),
  8. backbone=dict(
  9. norm_cfg=dict(requires_grad=False),
  10. style='caffe',
  11. init_cfg=dict(
  12. type='Pretrained',
  13. checkpoint='open-mmlab://detectron2/resnet50_caffe')),
  14. rpn_head=dict(
  15. loss_bbox=dict(type='SmoothL1Loss', beta=1.0 / 9.0, loss_weight=1.0)),
  16. roi_head=dict(
  17. bbox_roi_extractor=dict(
  18. roi_layer=dict(
  19. type='RoIAlign',
  20. output_size=7,
  21. sampling_ratio=2,
  22. aligned=False)),
  23. bbox_head=dict(
  24. loss_bbox=dict(type='SmoothL1Loss', beta=1.0, loss_weight=1.0)),
  25. mask_roi_extractor=dict(
  26. roi_layer=dict(
  27. type='RoIAlign',
  28. output_size=14,
  29. sampling_ratio=2,
  30. aligned=False))))