Nov3
Joseph 2020-10-28 10:22:02 +05:30
parent 77b239137a
commit c7be064638
1 changed files with 5 additions and 3 deletions

View File

@ -1,16 +1,18 @@
_BASE_: "../Base-RCNN-C4-OWOD.yaml"
MODEL:
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
# WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
WEIGHTS: "/home/fk1/workspace/OWOD/output/t1_20_class/model_final.pth"
# WEIGHTS: "/home/fk1/workspace/OWOD/output/t1/model_final.pth"
DATASETS:
TRAIN: ('voc_2007_trainval', 'voc_2012_trainval')
TEST: ('voc_2007_test', )
# TEST: ('t2_all_test_unk', )
TEST: ('voc_2007_test','t2_test_unk', 't3_test_unk', 't4_test_unk')
# TEST: ('voc_2007_test','t2_test_unk', 't3_test_unk', 't4_test_unk')
SOLVER:
STEPS: (12000, 16000)
MAX_ITER: 18000
WARMUP_ITERS: 100
OUTPUT_DIR: "./output/t1_20_class"
OUTPUT_DIR: "./output/t1_20_class_test"
OWOD:
PREV_INTRODUCED_CLS: 0
CUR_INTRODUCED_CLS: 20