| _BASE_: "../Base-RCNN-FPN.yaml" | |
| MODEL: | |
| WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-101.pkl" | |
| MASK_ON: True | |
| RESNETS: | |
| DEPTH: 101 | |
| ROI_HEADS: | |
| NUM_CLASSES: 1203 | |
| SCORE_THRESH_TEST: 0.0001 | |
| INPUT: | |
| MIN_SIZE_TRAIN: (640, 672, 704, 736, 768, 800) | |
| DATASETS: | |
| TRAIN: ("lvis_v1_train",) | |
| TEST: ("lvis_v1_val",) | |
| TEST: | |
| DETECTIONS_PER_IMAGE: 300 # LVIS allows up to 300 | |
| SOLVER: | |
| STEPS: (120000, 160000) | |
| MAX_ITER: 180000 # 180000 * 16 / 100000 ~ 28.8 epochs | |
| DATALOADER: | |
| SAMPLER_TRAIN: "RepeatFactorTrainingSampler" | |
| REPEAT_THRESHOLD: 0.001 | |