ld_r34-gflv1-r101_fpn_1x_coco.py 569 B

12345678910111213141516171819
  1. _base_ = ['./ld_r18-gflv1-r101_fpn_1x_coco.py']
  2. model = dict(
  3. backbone=dict(
  4. type='ResNet',
  5. depth=34,
  6. num_stages=4,
  7. out_indices=(0, 1, 2, 3),
  8. frozen_stages=1,
  9. norm_cfg=dict(type='BN', requires_grad=True),
  10. norm_eval=True,
  11. style='pytorch',
  12. init_cfg=dict(type='Pretrained', checkpoint='torchvision://resnet34')),
  13. neck=dict(
  14. type='FPN',
  15. in_channels=[64, 128, 256, 512],
  16. out_channels=256,
  17. start_level=1,
  18. add_extra_convs='on_output',
  19. num_outs=5))