faster_rcnn_R_50_FPN.yaml 449 B

123456789101112131415161718
  1. _BASE_: "../Base-RCNN-FPN.yaml"
  2. MODEL:
  3. WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
  4. MASK_ON: False
  5. RESNETS:
  6. DEPTH: 50
  7. ROI_HEADS:
  8. NUM_CLASSES: 20
  9. INPUT:
  10. MIN_SIZE_TRAIN: (480, 512, 544, 576, 608, 640, 672, 704, 736, 768, 800)
  11. MIN_SIZE_TEST: 800
  12. DATASETS:
  13. TRAIN: ('voc_2007_trainval', 'voc_2012_trainval')
  14. TEST: ('voc_2007_test',)
  15. SOLVER:
  16. STEPS: (12000, 16000)
  17. MAX_ITER: 18000 # 17.4 epochs
  18. WARMUP_ITERS: 100