RE-OWOD/configs/OWOD/t1/t1_train.yaml

17 lines
507 B
YAML

_BASE_: "../../Base-RCNN-C4-OWOD.yaml"
MODEL:
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
DATASETS:
TRAIN: ('t1_voc_coco_2007_train', ) # t1_voc_coco_2007_train, t1_voc_coco_2007_ft
TEST: ('voc_coco_2007_test', ) # voc_coco_2007_test, t1_voc_coco_2007_test, t1_voc_coco_2007_val
SOLVER:
STEPS: (60000, 80000)
MAX_ITER: 90000
WARMUP_ITERS: 100
OUTPUT_DIR: "./output/t1"
OWOD:
PREV_INTRODUCED_CLS: 0
CUR_INTRODUCED_CLS: 20
UNK_THRESH: 0
CALIBRATION: 0
GENERATE_CALI: False