mirror of https://github.com/JosephKJ/OWOD.git
Updates
parent
77b239137a
commit
c7be064638
|
@ -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
|
Loading…
Reference in New Issue