2020-09-29 08:17:18 +08:00
|
|
|
_BASE_: "../Base-RCNN-C4.yaml"
|
|
|
|
|
MODEL:
|
|
|
|
|
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
|
|
|
|
|
MASK_ON: False
|
|
|
|
|
RESNETS:
|
|
|
|
|
DEPTH: 50
|
|
|
|
|
ROI_HEADS:
|
2020-10-04 12:29:13 +08:00
|
|
|
NUM_CLASSES: 21 # 0-19 Known class; 20 -> Unknown; 21 -> Background.
|
|
|
|
|
# NUM_CLASSES: 20
|
2020-09-29 08:17:18 +08:00
|
|
|
INPUT:
|
|
|
|
|
MIN_SIZE_TRAIN: (480, 512, 544, 576, 608, 640, 672, 704, 736, 768, 800)
|
|
|
|
|
MIN_SIZE_TEST: 800
|
|
|
|
|
DATASETS:
|
2020-10-04 12:29:13 +08:00
|
|
|
|
2020-09-29 08:17:18 +08:00
|
|
|
TRAIN: ('voc_2007_trainval', 'voc_2012_trainval')
|
|
|
|
|
TEST: ('voc_2007_test',)
|
|
|
|
|
SOLVER:
|
|
|
|
|
STEPS: (12000, 16000)
|
|
|
|
|
MAX_ITER: 18000 # 17.4 epochs
|
|
|
|
|
WARMUP_ITERS: 100
|
2020-10-16 10:27:39 +08:00
|
|
|
OUTPUT_DIR: "./output/baseline_run"
|
2020-10-04 12:29:13 +08:00
|
|
|
OWOD:
|
2020-10-16 10:27:39 +08:00
|
|
|
ENABLE_THRESHOLD_AUTOLABEL_UNK: False
|
2020-10-04 12:29:13 +08:00
|
|
|
NUM_UNK_PER_IMAGE: 1
|
|
|
|
|
ENABLE_UNCERTAINITY_AUTOLABEL_UNK: False
|
2020-10-16 10:27:39 +08:00
|
|
|
ENABLE_CLUSTERING: False
|
2020-10-12 09:01:27 +08:00
|
|
|
CLUSTERING:
|
|
|
|
|
ITEMS_PER_CLASS: 20
|
2020-10-16 10:27:39 +08:00
|
|
|
START_ITER: 20
|
|
|
|
|
UPDATE_MU_ITER: 3000
|
2020-10-12 09:01:27 +08:00
|
|
|
MOMENTUM: 0.99
|
2020-10-16 10:27:39 +08:00
|
|
|
Z_DIMENSION: 128 # 64 128 is also 3.5 h
|
2020-10-12 09:01:27 +08:00
|
|
|
|
|
|
|
|
# CLUSTERING:
|
|
|
|
|
# ITEMS_PER_CLASS: 20
|
|
|
|
|
# START_ITER: 1000
|
|
|
|
|
# UPDATE_MU_ITER: 3000
|
|
|
|
|
# MOMENTUM: 0.99
|
