cascade-mask-rcnn_hrnetv2p-w18_20e_coco.py 462 B

1234567891011
  1. _base_ = './cascade-mask-rcnn_hrnetv2p-w32_20e_coco.py'
  2. # model settings
  3. model = dict(
  4. backbone=dict(
  5. extra=dict(
  6. stage2=dict(num_channels=(18, 36)),
  7. stage3=dict(num_channels=(18, 36, 72)),
  8. stage4=dict(num_channels=(18, 36, 72, 144))),
  9. init_cfg=dict(
  10. type='Pretrained', checkpoint='open-mmlab://msra/hrnetv2_w18')),
  11. neck=dict(type='HRFPN', in_channels=[18, 36, 72, 144], out_channels=256))