tridentnet_r50-caffe_1x_coco.py 748 B

12345678910111213141516171819202122
  1. _base_ = [
  2. '../_base_/models/faster-rcnn_r50-caffe-c4.py',
  3. '../_base_/datasets/coco_detection.py',
  4. '../_base_/schedules/schedule_1x.py', '../_base_/default_runtime.py'
  5. ]
  6. model = dict(
  7. type='TridentFasterRCNN',
  8. backbone=dict(
  9. type='TridentResNet',
  10. trident_dilations=(1, 2, 3),
  11. num_branch=3,
  12. test_branch_idx=1,
  13. init_cfg=dict(
  14. type='Pretrained',
  15. checkpoint='open-mmlab://detectron2/resnet50_caffe')),
  16. roi_head=dict(type='TridentRoIHead', num_branch=3, test_branch_idx=1),
  17. train_cfg=dict(
  18. rpn_proposal=dict(max_per_img=500),
  19. rcnn=dict(
  20. sampler=dict(num=128, pos_fraction=0.5,
  21. add_gt_as_proposals=False))))