htc_r50-rfp_1x_coco.py 714 B

123456789101112131415161718192021222324
  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. output_img=True),
  7. neck=dict(
  8. type='RFP',
  9. rfp_steps=2,
  10. aspp_out_channels=64,
  11. aspp_dilations=(1, 3, 6, 1),
  12. rfp_backbone=dict(
  13. rfp_inplanes=256,
  14. type='DetectoRS_ResNet',
  15. depth=50,
  16. num_stages=4,
  17. out_indices=(0, 1, 2, 3),
  18. frozen_stages=1,
  19. norm_cfg=dict(type='BN', requires_grad=True),
  20. norm_eval=True,
  21. conv_cfg=dict(type='ConvAWS'),
  22. pretrained='torchvision://resnet50',
  23. style='pytorch')))