faster-rcnn_r50_fpn_gn-ws-all_1x_coco.py 577 B

12345678910111213141516
  1. _base_ = '../faster_rcnn/faster-rcnn_r50_fpn_1x_coco.py'
  2. conv_cfg = dict(type='ConvWS')
  3. norm_cfg = dict(type='GN', num_groups=32, requires_grad=True)
  4. model = dict(
  5. backbone=dict(
  6. conv_cfg=conv_cfg,
  7. norm_cfg=norm_cfg,
  8. init_cfg=dict(
  9. type='Pretrained', checkpoint='open-mmlab://jhu/resnet50_gn_ws')),
  10. neck=dict(conv_cfg=conv_cfg, norm_cfg=norm_cfg),
  11. roi_head=dict(
  12. bbox_head=dict(
  13. type='Shared4Conv1FCBBoxHead',
  14. conv_out_channels=256,
  15. conv_cfg=conv_cfg,
  16. norm_cfg=norm_cfg)))