Base-RCNN-C4.yaml 368 B

123456789101112131415161718
  1. MODEL:
  2. META_ARCHITECTURE: "GeneralizedRCNN"
  3. RPN:
  4. PRE_NMS_TOPK_TEST: 6000
  5. POST_NMS_TOPK_TEST: 1000
  6. ROI_HEADS:
  7. NAME: "Res5ROIHeads"
  8. DATASETS:
  9. TRAIN: ("coco_2017_train",)
  10. TEST: ("coco_2017_val",)
  11. SOLVER:
  12. IMS_PER_BATCH: 16
  13. BASE_LR: 0.02
  14. STEPS: (60000, 80000)
  15. MAX_ITER: 90000
  16. INPUT:
  17. MIN_SIZE_TRAIN: (640, 672, 704, 736, 768, 800)
  18. VERSION: 2