Base-RCNN-DilatedC5.yaml 665 B

12345678910111213141516171819202122232425262728293031
  1. MODEL:
  2. META_ARCHITECTURE: "GeneralizedRCNN"
  3. RESNETS:
  4. OUT_FEATURES: ["res5"]
  5. RES5_DILATION: 2
  6. RPN:
  7. IN_FEATURES: ["res5"]
  8. PRE_NMS_TOPK_TEST: 6000
  9. POST_NMS_TOPK_TEST: 1000
  10. ROI_HEADS:
  11. NAME: "StandardROIHeads"
  12. IN_FEATURES: ["res5"]
  13. ROI_BOX_HEAD:
  14. NAME: "FastRCNNConvFCHead"
  15. NUM_FC: 2
  16. POOLER_RESOLUTION: 7
  17. ROI_MASK_HEAD:
  18. NAME: "MaskRCNNConvUpsampleHead"
  19. NUM_CONV: 4
  20. POOLER_RESOLUTION: 14
  21. DATASETS:
  22. TRAIN: ("coco_2017_train",)
  23. TEST: ("coco_2017_val",)
  24. SOLVER:
  25. IMS_PER_BATCH: 16
  26. BASE_LR: 0.02
  27. STEPS: (60000, 80000)
  28. MAX_ITER: 90000
  29. INPUT:
  30. MIN_SIZE_TRAIN: (640, 672, 704, 736, 768, 800)
  31. VERSION: 2