detectors_htc-r50_1x_coco.py 916 B

12345678910111213141516171819202122232425262728
  1. _base_ = '../htc/htc_r50_fpn_1x_coco.py'
  2. model = dict(
  3. backbone=dict(
  4. type='DetectoRS_ResNet',
  5. conv_cfg=dict(type='ConvAWS'),
  6. sac=dict(type='SAC', use_deform=True),
  7. stage_with_sac=(False, True, True, True),
  8. output_img=True),
  9. neck=dict(
  10. type='RFP',
  11. rfp_steps=2,
  12. aspp_out_channels=64,
  13. aspp_dilations=(1, 3, 6, 1),
  14. rfp_backbone=dict(
  15. rfp_inplanes=256,
  16. type='DetectoRS_ResNet',
  17. depth=50,
  18. num_stages=4,
  19. out_indices=(0, 1, 2, 3),
  20. frozen_stages=1,
  21. norm_cfg=dict(type='BN', requires_grad=True),
  22. norm_eval=True,
  23. conv_cfg=dict(type='ConvAWS'),
  24. sac=dict(type='SAC', use_deform=True),
  25. stage_with_sac=(False, True, True, True),
  26. pretrained='torchvision://resnet50',
  27. style='pytorch')))