mask-rcnn_r50_fpn_sample1e-3_ms-2x_lvis-v0.5.py 424 B

12345678910111213
  1. _base_ = [
  2. '../_base_/models/mask-rcnn_r50_fpn.py',
  3. '../_base_/datasets/lvis_v0.5_instance.py',
  4. '../_base_/schedules/schedule_2x.py', '../_base_/default_runtime.py'
  5. ]
  6. model = dict(
  7. roi_head=dict(
  8. bbox_head=dict(num_classes=1230), mask_head=dict(num_classes=1230)),
  9. test_cfg=dict(
  10. rcnn=dict(
  11. score_thr=0.0001,
  12. # LVIS allows up to 300
  13. max_per_img=300)))