faster-rcnn_r50_fpn_soft-nms_1x_coco.py 347 B

123456789101112
  1. _base_ = [
  2. '../_base_/models/faster-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. test_cfg=dict(
  8. rcnn=dict(
  9. score_thr=0.05,
  10. nms=dict(type='soft_nms', iou_threshold=0.5),
  11. max_per_img=100)))