mirror of https://github.com/RE-OWOD/RE-OWOD
16 lines
647 B
YAML
16 lines
647 B
YAML
_BASE_: "../Base-RCNN-C4-OWOD.yaml"
|
|
MODEL:
|
|
WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
|
|
# WEIGHTS: "/home/fk1/workspace/OWOD/output/expr_training_with_unk_with_clustering_Z_DIMENSION_256/model_final.pth"
|
|
# WEIGHTS: "/home/fk1/workspace/OWOD/output/t1_std_frcnn/model_final.pth"
|
|
DATASETS:
|
|
TRAIN: ('t1_voc_coco_2007_train', ) # t1_voc_coco_2007_train, t1_voc_coco_2007_ft
|
|
TEST: ('voc_coco_2007_test', ) # t1_voc_coco_2007_test, t1_voc_coco_2007_val
|
|
SOLVER:
|
|
STEPS: (12000, 16000)
|
|
MAX_ITER: 18000
|
|
WARMUP_ITERS: 100
|
|
OUTPUT_DIR: "./output/t1_ENABLE_CLUSTERING_margin_5"
|
|
OWOD:
|
|
PREV_INTRODUCED_CLS: 0
|
|
CUR_INTRODUCED_CLS: 20 |