27 lines
693 B
YAML
27 lines
693 B
YAML
_BASE_: "../Base-RCNN-FPN.yaml"
|
|
MODEL:
|
|
WEIGHTS: "detectron2://ImageNetPretrained/FAIR/X-101-32x8d.pkl"
|
|
PIXEL_STD: [57.375, 57.120, 58.395]
|
|
MASK_ON: True
|
|
RESNETS:
|
|
STRIDE_IN_1X1: False # this is a C2 model
|
|
NUM_GROUPS: 32
|
|
WIDTH_PER_GROUP: 8
|
|
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",)
|
|
SOLVER:
|
|
STEPS: (120000, 160000)
|
|
MAX_ITER: 180000 # 180000 * 16 / 100000 ~ 28.8 epochs
|
|
TEST:
|
|
DETECTIONS_PER_IMAGE: 300 # LVIS allows up to 300
|
|
DATALOADER:
|
|
SAMPLER_TRAIN: "RepeatFactorTrainingSampler"
|
|
REPEAT_THRESHOLD: 0.001
|