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" _BASE_: "../Base-RCNN-C4-OWOD.yaml"
MODEL: 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" # WEIGHTS: "/home/fk1/workspace/OWOD/output/t1/model_final.pth"
DATASETS: DATASETS:
TRAIN: ('voc_2007_trainval', 'voc_2012_trainval') TRAIN: ('voc_2007_trainval', 'voc_2012_trainval')
TEST: ('voc_2007_test', )
# TEST: ('t2_all_test_unk', ) # 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: SOLVER:
STEPS: (12000, 16000) STEPS: (12000, 16000)
MAX_ITER: 18000 MAX_ITER: 18000
WARMUP_ITERS: 100 WARMUP_ITERS: 100
OUTPUT_DIR: "./output/t1_20_class" OUTPUT_DIR: "./output/t1_20_class_test"
OWOD: OWOD:
PREV_INTRODUCED_CLS: 0 PREV_INTRODUCED_CLS: 0
CUR_INTRODUCED_CLS: 20 CUR_INTRODUCED_CLS: 20