cascade-rcnn_r50-rfp_1x_coco.py 851 B

12345678910111213141516171819202122232425262728
  1. _base_ = [
  2. '../_base_/models/cascade-rcnn_r50_fpn.py',
  3. '../_base_/datasets/coco_detection.py',
  4. '../_base_/schedules/schedule_1x.py', '../_base_/default_runtime.py'
  5. ]
  6. model = dict(
  7. backbone=dict(
  8. type='DetectoRS_ResNet',
  9. conv_cfg=dict(type='ConvAWS'),
  10. output_img=True),
  11. neck=dict(
  12. type='RFP',
  13. rfp_steps=2,
  14. aspp_out_channels=64,
  15. aspp_dilations=(1, 3, 6, 1),
  16. rfp_backbone=dict(
  17. rfp_inplanes=256,
  18. type='DetectoRS_ResNet',
  19. depth=50,
  20. num_stages=4,
  21. out_indices=(0, 1, 2, 3),
  22. frozen_stages=1,
  23. norm_cfg=dict(type='BN', requires_grad=True),
  24. norm_eval=True,
  25. conv_cfg=dict(type='ConvAWS'),
  26. pretrained='torchvision://resnet50',
  27. style='pytorch')))