retinanet_r50_nasfpn_crop640-50e_coco.py 480 B

12345678910111213141516
  1. _base_ = './retinanet_r50_fpn_crop640-50e_coco.py'
  2. # model settings
  3. model = dict(
  4. # `pad_size_divisor=128` ensures the feature maps sizes
  5. # in `NAS_FPN` won't mismatch.
  6. data_preprocessor=dict(pad_size_divisor=128),
  7. neck=dict(
  8. _delete_=True,
  9. type='NASFPN',
  10. in_channels=[256, 512, 1024, 2048],
  11. out_channels=256,
  12. num_outs=5,
  13. stack_times=7,
  14. start_level=1,
  15. norm_cfg=dict(type='BN', requires_grad=True)))