Add files via upload

configs/Base-RCNN-C4-OWOD.yaml

@@ -0,0 +1,40 @@
+MODEL:
+  META_ARCHITECTURE: "GeneralizedRCNN"
+  RPN:
+    PRE_NMS_TOPK_TEST: 6000
+    POST_NMS_TOPK_TEST: 1000
+  ROI_HEADS:
+    #    NUM_CLASSES: 81 # 0-79 Known class; 80 -> Unknown; 81 -> Background.
+    NUM_CLASSES: 81
+    NAME: "Res5ROIHeads"
+  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
+  MASK_ON: False
+  RESNETS:
+    DEPTH: 50
+INPUT:
+  MIN_SIZE_TRAIN: (480, 512, 544, 576, 608, 640, 672, 704, 736, 768, 800)
+  MIN_SIZE_TEST: 800
+DATASETS:
+  TRAIN: ("coco_2017_train",)
+  TEST: ("coco_2017_val",)
+SOLVER:
+  IMS_PER_BATCH: 16
+  BASE_LR: 0.02
+  STEPS: (60000, 80000)
+  MAX_ITER: 90000
+VERSION: 2
+OWOD:
+  ENABLE_THRESHOLD_AUTOLABEL_UNK: True
+  NUM_UNK_PER_IMAGE: 1
+  ENABLE_UNCERTAINITY_AUTOLABEL_UNK: False
+  ENABLE_CLUSTERING: True
+  FEATURE_STORE_SAVE_PATH: 'feature_store'
+  CLUSTERING:
+    ITEMS_PER_CLASS: 20
+    START_ITER: 0 #10000
+    UPDATE_MU_ITER: 1
+    MOMENTUM: 0.99
+    Z_DIMENSION: 128
+    UNK_THRESH: 0.7
+    CALIBRATION: 0.7
+    GENERATE_CALI: False

configs/Base-RCNN-C4-OWOD_FPN.yaml

@@ -0,0 +1,53 @@
+MODEL:
+  META_ARCHITECTURE: "GeneralizedRCNN"
+  RPN:
+    IN_FEATURES: ["p2", "p3", "p4", "p5", "p6"]
+    PRE_NMS_TOPK_TEST: 6000
+    POST_NMS_TOPK_TEST: 1000
+  ROI_HEADS:
+    #    NUM_CLASSES: 81 # 0-79 Known class; 80 -> Unknown; 81 -> Background.
+    NUM_CLASSES: 81
+    IN_FEATURES: ["p2", "p3", "p4", "p5"]
+    NAME: "Res5ROIHeads"
+  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
+  MASK_ON: False
+  RESNETS:
+    DEPTH: 50
+    OUT_FEATURES: ["res2", "res3", "res4", "res5"]
+  FPN:
+    IN_FEATURES: ["res2", "res3", "res4", "res5"]
+  ANCHOR_GENERATOR:
+    SIZES: [[32], [64], [128], [256], [512]]  # One size for each in feature map
+    ASPECT_RATIOS: [[0.5, 1.0, 2.0]]  # Three aspect ratios (same for all in feature maps
+  ROI_BOX_HEAD:
+    NAME: "FastRCNNConvFCHead"
+    NUM_FC: 2
+    POOLER_RESOLUTION: 7
+  ROI_MASK_HEAD:
+    NAME: "MaskRCNNConvUpsampleHead"
+    NUM_CONV: 4
+    POOLER_RESOLUTION: 14
+INPUT:
+  MIN_SIZE_TRAIN: (480, 512, 544, 576, 608, 640, 672, 704, 736, 768, 800)
+  MIN_SIZE_TEST: 800
+DATASETS:
+  TRAIN: ("coco_2017_train",)
+  TEST: ("coco_2017_val",)
+SOLVER:
+  IMS_PER_BATCH: 16
+  BASE_LR: 0.02
+  STEPS: (60000, 80000)
+  MAX_ITER: 90000
+VERSION: 2
+OWOD:
+  ENABLE_THRESHOLD_AUTOLABEL_UNK: True
+  NUM_UNK_PER_IMAGE: 1
+  ENABLE_UNCERTAINITY_AUTOLABEL_UNK: False
+  ENABLE_CLUSTERING: True
+  FEATURE_STORE_SAVE_PATH: 'feature_store'
+  CLUSTERING:
+    ITEMS_PER_CLASS: 20
+    START_ITER: 0 #1000
+    UPDATE_MU_ITER: 3000
+    MOMENTUM: 0.99
+    Z_DIMENSION: 128

configs/Base-RCNN-C4.yaml

@@ -0,0 +1,18 @@
+MODEL:
+  META_ARCHITECTURE: "GeneralizedRCNN"
+  RPN:
+    PRE_NMS_TOPK_TEST: 6000
+    POST_NMS_TOPK_TEST: 1000
+  ROI_HEADS:
+    NAME: "Res5ROIHeads"
+DATASETS:
+  TRAIN: ("coco_2017_train",)
+  TEST: ("coco_2017_val",)
+SOLVER:
+  IMS_PER_BATCH: 16
+  BASE_LR: 0.02
+  STEPS: (60000, 80000)
+  MAX_ITER: 90000
+INPUT:
+  MIN_SIZE_TRAIN: (640, 672, 704, 736, 768, 800)
+VERSION: 2

configs/Base-RCNN-DilatedC5.yaml

@@ -0,0 +1,31 @@
+MODEL:
+  META_ARCHITECTURE: "GeneralizedRCNN"
+  RESNETS:
+    OUT_FEATURES: ["res5"]
+    RES5_DILATION: 2
+  RPN:
+    IN_FEATURES: ["res5"]
+    PRE_NMS_TOPK_TEST: 6000
+    POST_NMS_TOPK_TEST: 1000
+  ROI_HEADS:
+    NAME: "StandardROIHeads"
+    IN_FEATURES: ["res5"]
+  ROI_BOX_HEAD:
+    NAME: "FastRCNNConvFCHead"
+    NUM_FC: 2
+    POOLER_RESOLUTION: 7
+  ROI_MASK_HEAD:
+    NAME: "MaskRCNNConvUpsampleHead"
+    NUM_CONV: 4
+    POOLER_RESOLUTION: 14
+DATASETS:
+  TRAIN: ("coco_2017_train",)
+  TEST: ("coco_2017_val",)
+SOLVER:
+  IMS_PER_BATCH: 16
+  BASE_LR: 0.02
+  STEPS: (60000, 80000)
+  MAX_ITER: 90000
+INPUT:
+  MIN_SIZE_TRAIN: (640, 672, 704, 736, 768, 800)
+VERSION: 2

configs/Base-RCNN-FPN.yaml

@@ -0,0 +1,42 @@
+MODEL:
+  META_ARCHITECTURE: "GeneralizedRCNN"
+  BACKBONE:
+    NAME: "build_resnet_fpn_backbone"
+  RESNETS:
+    OUT_FEATURES: ["res2", "res3", "res4", "res5"]
+  FPN:
+    IN_FEATURES: ["res2", "res3", "res4", "res5"]
+  ANCHOR_GENERATOR:
+    SIZES: [[32], [64], [128], [256], [512]]  # One size for each in feature map
+    ASPECT_RATIOS: [[0.5, 1.0, 2.0]]  # Three aspect ratios (same for all in feature maps)
+  RPN:
+    IN_FEATURES: ["p2", "p3", "p4", "p5", "p6"]
+    PRE_NMS_TOPK_TRAIN: 2000  # Per FPN level
+    PRE_NMS_TOPK_TEST: 1000  # Per FPN level
+    # Detectron1 uses 2000 proposals per-batch,
+    # (See "modeling/rpn/rpn_outputs.py" for details of this legacy issue)
+    # which is approximately 1000 proposals per-image since the default batch size for FPN is 2.
+    POST_NMS_TOPK_TRAIN: 1000
+    POST_NMS_TOPK_TEST: 1000
+  ROI_HEADS:
+    NAME: "StandardROIHeads"
+    IN_FEATURES: ["p2", "p3", "p4", "p5"]
+  ROI_BOX_HEAD:
+    NAME: "FastRCNNConvFCHead"
+    NUM_FC: 2
+    POOLER_RESOLUTION: 7
+  ROI_MASK_HEAD:
+    NAME: "MaskRCNNConvUpsampleHead"
+    NUM_CONV: 4
+    POOLER_RESOLUTION: 14
+DATASETS:
+  TRAIN: ("coco_2017_train",)
+  TEST: ("coco_2017_val",)
+SOLVER:
+  IMS_PER_BATCH: 16
+  BASE_LR: 0.02
+  STEPS: (60000, 80000)
+  MAX_ITER: 90000
+INPUT:
+  MIN_SIZE_TRAIN: (640, 672, 704, 736, 768, 800)
+VERSION: 2

configs/Base-RetinaNet.yaml

@@ -0,0 +1,25 @@
+MODEL:
+  META_ARCHITECTURE: "RetinaNet"
+  BACKBONE:
+    NAME: "build_retinanet_resnet_fpn_backbone"
+  RESNETS:
+    OUT_FEATURES: ["res3", "res4", "res5"]
+  ANCHOR_GENERATOR:
+    SIZES: !!python/object/apply:eval ["[[x, x * 2**(1.0/3), x * 2**(2.0/3) ] for x in [32, 64, 128, 256, 512 ]]"]
+  FPN:
+    IN_FEATURES: ["res3", "res4", "res5"]
+  RETINANET:
+    IOU_THRESHOLDS: [0.4, 0.5]
+    IOU_LABELS: [0, -1, 1]
+    SMOOTH_L1_LOSS_BETA: 0.0
+DATASETS:
+  TRAIN: ("coco_2017_train",)
+  TEST: ("coco_2017_val",)
+SOLVER:
+  IMS_PER_BATCH: 16
+  BASE_LR: 0.01  # Note that RetinaNet uses a different default learning rate
+  STEPS: (60000, 80000)
+  MAX_ITER: 90000
+INPUT:
+  MIN_SIZE_TRAIN: (640, 672, 704, 736, 768, 800)
+VERSION: 2

configs/COCO-Detection/fast_rcnn_R_50_FPN_1x.yaml

@@ -0,0 +1,17 @@
+_BASE_: "../Base-RCNN-FPN.yaml"
+MODEL:
+  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
+  MASK_ON: False
+  LOAD_PROPOSALS: True
+  RESNETS:
+    DEPTH: 50
+  PROPOSAL_GENERATOR:
+    NAME: "PrecomputedProposals"
+DATASETS:
+  TRAIN: ("coco_2017_train",)
+  PROPOSAL_FILES_TRAIN: ("detectron2://COCO-Detection/rpn_R_50_FPN_1x/137258492/coco_2017_train_box_proposals_21bc3a.pkl", )
+  TEST: ("coco_2017_val",)
+  PROPOSAL_FILES_TEST: ("detectron2://COCO-Detection/rpn_R_50_FPN_1x/137258492/coco_2017_val_box_proposals_ee0dad.pkl", )
+DATALOADER:
+  # proposals are part of the dataset_dicts, and take a lot of RAM
+  NUM_WORKERS: 2

configs/COCO-Detection/faster_rcnn_R_101_C4_3x.yaml

@@ -0,0 +1,9 @@
+_BASE_: "../Base-RCNN-C4.yaml"
+MODEL:
+  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-101.pkl"
+  MASK_ON: False
+  RESNETS:
+    DEPTH: 101
+SOLVER:
+  STEPS: (210000, 250000)
+  MAX_ITER: 270000

configs/COCO-Detection/faster_rcnn_R_101_DC5_3x.yaml

@@ -0,0 +1,9 @@
+_BASE_: "../Base-RCNN-DilatedC5.yaml"
+MODEL:
+  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-101.pkl"
+  MASK_ON: False
+  RESNETS:
+    DEPTH: 101
+SOLVER:
+  STEPS: (210000, 250000)
+  MAX_ITER: 270000

configs/COCO-Detection/faster_rcnn_R_101_FPN_3x.yaml

@@ -0,0 +1,9 @@
+_BASE_: "../Base-RCNN-FPN.yaml"
+MODEL:
+  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-101.pkl"
+  MASK_ON: False
+  RESNETS:
+    DEPTH: 101
+SOLVER:
+  STEPS: (210000, 250000)
+  MAX_ITER: 270000

configs/COCO-Detection/faster_rcnn_R_50_C4_1x.yaml

@@ -0,0 +1,6 @@
+_BASE_: "../Base-RCNN-C4.yaml"
+MODEL:
+  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
+  MASK_ON: False
+  RESNETS:
+    DEPTH: 50

configs/COCO-Detection/faster_rcnn_R_50_C4_3x.yaml

@@ -0,0 +1,9 @@
+_BASE_: "../Base-RCNN-C4.yaml"
+MODEL:
+  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
+  MASK_ON: False
+  RESNETS:
+    DEPTH: 50
+SOLVER:
+  STEPS: (210000, 250000)
+  MAX_ITER: 270000

configs/COCO-Detection/faster_rcnn_R_50_DC5_1x.yaml

@@ -0,0 +1,6 @@
+_BASE_: "../Base-RCNN-DilatedC5.yaml"
+MODEL:
+  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
+  MASK_ON: False
+  RESNETS:
+    DEPTH: 50

configs/COCO-Detection/faster_rcnn_R_50_DC5_3x.yaml

@@ -0,0 +1,9 @@
+_BASE_: "../Base-RCNN-DilatedC5.yaml"
+MODEL:
+  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
+  MASK_ON: False
+  RESNETS:
+    DEPTH: 50
+SOLVER:
+  STEPS: (210000, 250000)
+  MAX_ITER: 270000

configs/COCO-Detection/faster_rcnn_R_50_FPN_1x.yaml

@@ -0,0 +1,6 @@
+_BASE_: "../Base-RCNN-FPN.yaml"
+MODEL:
+  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
+  MASK_ON: False
+  RESNETS:
+    DEPTH: 50

configs/COCO-Detection/faster_rcnn_R_50_FPN_3x.yaml

@@ -0,0 +1,9 @@
+_BASE_: "../Base-RCNN-FPN.yaml"
+MODEL:
+  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
+  MASK_ON: False
+  RESNETS:
+    DEPTH: 50
+SOLVER:
+  STEPS: (210000, 250000)
+  MAX_ITER: 270000

configs/COCO-Detection/faster_rcnn_X_101_32x8d_FPN_3x.yaml

@@ -0,0 +1,13 @@
+_BASE_: "../Base-RCNN-FPN.yaml"
+MODEL:
+  MASK_ON: False
+  WEIGHTS: "detectron2://ImageNetPretrained/FAIR/X-101-32x8d.pkl"
+  PIXEL_STD: [57.375, 57.120, 58.395]
+  RESNETS:
+    STRIDE_IN_1X1: False  # this is a C2 model
+    NUM_GROUPS: 32
+    WIDTH_PER_GROUP: 8
+    DEPTH: 101
+SOLVER:
+  STEPS: (210000, 250000)
+  MAX_ITER: 270000

configs/COCO-Detection/retinanet_R_101_FPN_3x.yaml

@@ -0,0 +1,8 @@
+_BASE_: "../Base-RetinaNet.yaml"
+MODEL:
+  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-101.pkl"
+  RESNETS:
+    DEPTH: 101
+SOLVER:
+  STEPS: (210000, 250000)
+  MAX_ITER: 270000

configs/COCO-Detection/retinanet_R_50_FPN_1x.yaml

@@ -0,0 +1,5 @@
+_BASE_: "../Base-RetinaNet.yaml"
+MODEL:
+  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
+  RESNETS:
+    DEPTH: 50

configs/COCO-Detection/retinanet_R_50_FPN_3x.yaml

@@ -0,0 +1,8 @@
+_BASE_: "../Base-RetinaNet.yaml"
+MODEL:
+  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
+  RESNETS:
+    DEPTH: 50
+SOLVER:
+  STEPS: (210000, 250000)
+  MAX_ITER: 270000

configs/COCO-Detection/rpn_R_50_C4_1x.yaml

@@ -0,0 +1,10 @@
+_BASE_: "../Base-RCNN-C4.yaml"
+MODEL:
+  META_ARCHITECTURE: "ProposalNetwork"
+  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
+  MASK_ON: False
+  RESNETS:
+    DEPTH: 50
+  RPN:
+    PRE_NMS_TOPK_TEST: 12000
+    POST_NMS_TOPK_TEST: 2000

configs/COCO-Detection/rpn_R_50_FPN_1x.yaml

@@ -0,0 +1,9 @@
+_BASE_: "../Base-RCNN-FPN.yaml"
+MODEL:
+  META_ARCHITECTURE: "ProposalNetwork"
+  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
+  MASK_ON: False
+  RESNETS:
+    DEPTH: 50
+  RPN:
+    POST_NMS_TOPK_TEST: 2000

configs/OWOD/iOD/10_p_10/base_10_train.yaml

@@ -0,0 +1,23 @@
+_BASE_: "../../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
+#  WEIGHTS: "/home/fk1/workspace/OWOD/output/iOD/base_10/model_final.pth"
+  ROI_HEADS:
+    #    NUM_CLASSES: 21 # 0-19 Known class; 20 -> Unknown; 21 -> Background.
+    NUM_CLASSES: 21
+  ROI_BOX_HEAD:
+    CLS_AGNOSTIC_BBOX_REG: True
+DATASETS:
+  TRAIN: ('voc_2007_trainval', )
+  TEST: ('voc_2007_test', )
+SOLVER:
+  STEPS: (12000, 16000)
+  MAX_ITER: 18000
+  WARMUP_ITERS: 100
+OUTPUT_DIR: "./output/iOD/base_10"
+OWOD:
+  ENABLE_THRESHOLD_AUTOLABEL_UNK: False
+  ENABLE_CLUSTERING: False
+  PREV_INTRODUCED_CLS: 0
+  CUR_INTRODUCED_CLS: 10
+SEED: 9999

configs/OWOD/iOD/10_p_10/ft.yaml

@@ -0,0 +1,21 @@
+_BASE_: "../../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+#  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
+  WEIGHTS: "/home/fk1/workspace/OWOD/output/iOD/10_p_10_incr_ft/model_final.pth"
+  ROI_HEADS:
+    #    NUM_CLASSES: 21 # 0-19 Known class; 20 -> Unknown; 21 -> Background.
+    NUM_CLASSES: 21
+DATASETS:
+  TRAIN: ('voc_2007_trn_ft', )
+  TEST: ('voc_2007_test', )
+SOLVER:
+  STEPS: (48000,52000)
+  MAX_ITER: 54000
+  WARMUP_ITERS: 100
+  CHECKPOINT_PERIOD: 1000
+OUTPUT_DIR: "./output/iOD/10_p_10_incr_ft"
+OWOD:
+  ENABLE_THRESHOLD_AUTOLABEL_UNK: False
+  ENABLE_CLUSTERING: False
+  PREV_INTRODUCED_CLS: 10
+  CUR_INTRODUCED_CLS: 10

configs/OWOD/iOD/10_p_10/ft_with_unk.yaml

@@ -0,0 +1,23 @@
+_BASE_: "../../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+#  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
+  WEIGHTS: "/home/fk1/workspace/OWOD/output/iOD/10_p_10_ft_10_per_class/model_final.pth"
+  ROI_HEADS:
+    #    NUM_CLASSES: 21 # 0-19 Known class; 20 -> Unknown; 21 -> Background.
+    NUM_CLASSES: 21
+  ROI_BOX_HEAD:
+    CLS_AGNOSTIC_BBOX_REG: True
+DATASETS:
+  TRAIN: ('voc_2007_trn_ft', )
+  TEST: ('voc_2007_test', )
+SOLVER:
+  STEPS: (32000,34000)
+  MAX_ITER: 36000
+  WARMUP_ITERS: 100
+OUTPUT_DIR: "./output/iOD/10_p_10_ft_10_per_class"
+OWOD:
+  ENABLE_THRESHOLD_AUTOLABEL_UNK: False
+  ENABLE_CLUSTERING: False
+  PREV_INTRODUCED_CLS: 10
+  CUR_INTRODUCED_CLS: 10
+SEED: 9999

configs/OWOD/iOD/10_p_10/next_10_train.yaml

@@ -0,0 +1,20 @@
+_BASE_: "../../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+#  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
+  WEIGHTS: "/home/fk1/workspace/OWOD/output/iOD/10_p_10_incr/model_final.pth"
+  ROI_HEADS:
+    #    NUM_CLASSES: 21 # 0-19 Known class; 20 -> Unknown; 21 -> Background.
+    NUM_CLASSES: 21
+DATASETS:
+  TRAIN: ('voc_2007_train', )
+  TEST: ('voc_2007_test', )
+SOLVER:
+  STEPS: (32000, 34000)
+  MAX_ITER: 36000
+  WARMUP_ITERS: 100
+OUTPUT_DIR: "./output/iOD/10_p_10_incr"
+OWOD:
+  ENABLE_THRESHOLD_AUTOLABEL_UNK: False
+  ENABLE_CLUSTERING: False
+  PREV_INTRODUCED_CLS: 10
+  CUR_INTRODUCED_CLS: 10

configs/OWOD/iOD/10_p_10/next_10_train_with_unk_detection.yaml

@@ -0,0 +1,26 @@
+_BASE_: "../../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+#  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
+  WEIGHTS: "/home/fk1/workspace/OWOD/output/iOD/10_p_10/model_0009999.pth"
+  ROI_HEADS:
+    #    NUM_CLASSES: 21 # 0-19 Known class; 20 -> Unknown; 21 -> Background.
+    NUM_CLASSES: 21
+  ROI_BOX_HEAD:
+    CLS_AGNOSTIC_BBOX_REG: True
+DATASETS:
+  TRAIN: ('voc_2007_trainval', )
+  TEST: ('voc_2007_test', )
+SOLVER:
+  STEPS: (22000, 24000)
+  MAX_ITER: 26000
+  WARMUP_ITERS: 100
+OUTPUT_DIR: "./output/iOD/10_p_10"
+OWOD:
+  ENABLE_THRESHOLD_AUTOLABEL_UNK: True
+  ENABLE_CLUSTERING: False
+  PREV_INTRODUCED_CLS: 10
+  CUR_INTRODUCED_CLS: 10
+  CLUSTERING:
+    START_ITER: 20000
+    UPDATE_MU_ITER: 3100
+SEED: 9999

configs/OWOD/iOD/15_p_5/base_15_train.yaml

@@ -0,0 +1,23 @@
+_BASE_: "../../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
+#  WEIGHTS: "/home/fk1/workspace/OWOD/output/iOD/base_10/model_final.pth"
+  ROI_HEADS:
+    #    NUM_CLASSES: 21 # 0-19 Known class; 20 -> Unknown; 21 -> Background.
+    NUM_CLASSES: 21
+  ROI_BOX_HEAD:
+    CLS_AGNOSTIC_BBOX_REG: True
+DATASETS:
+  TRAIN: ('voc_2007_trainval', )
+  TEST: ('voc_2007_test', )
+SOLVER:
+  STEPS: (12000, 16000)
+  MAX_ITER: 18000
+  WARMUP_ITERS: 100
+OUTPUT_DIR: "./output/iOD/base_15"
+OWOD:
+  ENABLE_THRESHOLD_AUTOLABEL_UNK: False
+  ENABLE_CLUSTERING: False
+  PREV_INTRODUCED_CLS: 0
+  CUR_INTRODUCED_CLS: 15
+SEED: 9999

configs/OWOD/iOD/15_p_5/ft_with_unk.yaml

@@ -0,0 +1,23 @@
+_BASE_: "../../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+#  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
+  WEIGHTS: "/home/fk1/workspace/OWOD/output/iOD/15_p_5_ft/model_final.pth"
+  ROI_HEADS:
+    #    NUM_CLASSES: 21 # 0-19 Known class; 20 -> Unknown; 21 -> Background.
+    NUM_CLASSES: 21
+  ROI_BOX_HEAD:
+    CLS_AGNOSTIC_BBOX_REG: True
+DATASETS:
+  TRAIN: ('voc_2007_trn_ft', )
+  TEST: ('voc_2007_test', )
+SOLVER:
+  STEPS: (32000,34000)
+  MAX_ITER: 36000
+  WARMUP_ITERS: 100
+OUTPUT_DIR: "./output/iOD/15_p_5_ft"
+OWOD:
+  ENABLE_THRESHOLD_AUTOLABEL_UNK: False
+  ENABLE_CLUSTERING: False
+  PREV_INTRODUCED_CLS: 15
+  CUR_INTRODUCED_CLS: 5
+SEED: 9999

configs/OWOD/iOD/15_p_5/next_5_train_with_ud.yaml

@@ -0,0 +1,26 @@
+_BASE_: "../../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+#  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
+  WEIGHTS: "/home/fk1/workspace/OWOD/output/iOD/15_p_5/model_0009999.pth"
+  ROI_HEADS:
+    #    NUM_CLASSES: 21 # 0-19 Known class; 20 -> Unknown; 21 -> Background.
+    NUM_CLASSES: 21
+  ROI_BOX_HEAD:
+    CLS_AGNOSTIC_BBOX_REG: True
+DATASETS:
+  TRAIN: ('voc_2007_trainval', )
+  TEST: ('voc_2007_test', )
+SOLVER:
+  STEPS: (22000, 24000)
+  MAX_ITER: 20000
+  WARMUP_ITERS: 100
+OUTPUT_DIR: "./output/iOD/15_p_5"
+OWOD:
+  ENABLE_THRESHOLD_AUTOLABEL_UNK: True
+  ENABLE_CLUSTERING: False
+  PREV_INTRODUCED_CLS: 15
+  CUR_INTRODUCED_CLS: 5
+  CLUSTERING:
+    START_ITER: 20000
+    UPDATE_MU_ITER: 3100
+SEED: 9999

configs/OWOD/iOD/19_p_1/base_19_train.yaml

@@ -0,0 +1,23 @@
+_BASE_: "../../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
+#  WEIGHTS: "/home/fk1/workspace/OWOD/output/iOD/base_10/model_final.pth"
+  ROI_HEADS:
+    #    NUM_CLASSES: 21 # 0-19 Known class; 20 -> Unknown; 21 -> Background.
+    NUM_CLASSES: 21
+  ROI_BOX_HEAD:
+    CLS_AGNOSTIC_BBOX_REG: True
+DATASETS:
+  TRAIN: ('voc_2007_trainval', )
+  TEST: ('voc_2007_test', )
+SOLVER:
+  STEPS: (12000, 16000)
+  MAX_ITER: 18000
+  WARMUP_ITERS: 100
+OUTPUT_DIR: "./output/iOD/base_19"
+OWOD:
+  ENABLE_THRESHOLD_AUTOLABEL_UNK: False
+  ENABLE_CLUSTERING: False
+  PREV_INTRODUCED_CLS: 0
+  CUR_INTRODUCED_CLS: 19
+SEED: 9999

configs/OWOD/iOD/19_p_1/ft_with_unk.yaml

@@ -0,0 +1,23 @@
+_BASE_: "../../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+#  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
+  WEIGHTS: "/home/fk1/workspace/OWOD/output/iOD/19_p_1_ft/model_final.pth"
+  ROI_HEADS:
+    #    NUM_CLASSES: 21 # 0-19 Known class; 20 -> Unknown; 21 -> Background.
+    NUM_CLASSES: 21
+  ROI_BOX_HEAD:
+    CLS_AGNOSTIC_BBOX_REG: True
+DATASETS:
+  TRAIN: ('voc_2007_trn_ft', )
+  TEST: ('voc_2007_test', )
+SOLVER:
+  STEPS: (32000,34000)
+  MAX_ITER: 36000
+  WARMUP_ITERS: 100
+OUTPUT_DIR: "./output/iOD/19_p_1_ft"
+OWOD:
+  ENABLE_THRESHOLD_AUTOLABEL_UNK: False
+  ENABLE_CLUSTERING: False
+  PREV_INTRODUCED_CLS: 19
+  CUR_INTRODUCED_CLS: 1
+SEED: 9999

configs/OWOD/iOD/19_p_1/next_1_train_with_ud.yaml

@@ -0,0 +1,26 @@
+_BASE_: "../../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+#  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
+  WEIGHTS: "/home/fk1/workspace/OWOD/output/iOD/19_p_1/model_0009999.pth"
+  ROI_HEADS:
+    #    NUM_CLASSES: 21 # 0-19 Known class; 20 -> Unknown; 21 -> Background.
+    NUM_CLASSES: 21
+  ROI_BOX_HEAD:
+    CLS_AGNOSTIC_BBOX_REG: True
+DATASETS:
+  TRAIN: ('voc_2007_trainval', )
+  TEST: ('voc_2007_test', )
+SOLVER:
+  STEPS: (22000, 24000)
+  MAX_ITER: 20000
+  WARMUP_ITERS: 100
+OUTPUT_DIR: "./output/iOD/19_p_1"
+OWOD:
+  ENABLE_THRESHOLD_AUTOLABEL_UNK: True
+  ENABLE_CLUSTERING: False
+  PREV_INTRODUCED_CLS: 19
+  CUR_INTRODUCED_CLS: 1
+  CLUSTERING:
+    START_ITER: 20000
+    UPDATE_MU_ITER: 3100
+SEED: 9999

configs/OWOD/iOD/all_20_train.yaml

@@ -0,0 +1,23 @@
+_BASE_: "../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
+#  WEIGHTS: "/home/fk1/workspace/OWOD/output/iOD/all_20/model_final.pth"
+  ROI_HEADS:
+    #    NUM_CLASSES: 21 # 0-19 Known class; 20 -> Unknown; 21 -> Background.
+    NUM_CLASSES: 21
+  ROI_BOX_HEAD:
+    CLS_AGNOSTIC_BBOX_REG: True
+DATASETS:
+  TRAIN: ('voc_2007_trainval', )
+  TEST: ('voc_2007_test', )
+SOLVER:
+  STEPS: (12000, 16000)
+  MAX_ITER: 18000
+  WARMUP_ITERS: 100
+OUTPUT_DIR: "./output/iOD/all_20"
+OWOD:
+  ENABLE_THRESHOLD_AUTOLABEL_UNK: False
+  ENABLE_CLUSTERING: False
+  PREV_INTRODUCED_CLS: 0
+  CUR_INTRODUCED_CLS: 20
+SEED: 9999

configs/OWOD/iOD/test.yaml

@@ -0,0 +1,30 @@
+_BASE_: "../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+#  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
+#  WEIGHTS: "/home/fk1/workspace/OWOD/output/iOD/base_19/model_final.pth"
+  WEIGHTS: "/home/fk1/workspace/OWOD/output/iOD/10_p_10_ft/model_0024999.pth"
+#  WEIGHTS: "/home/fk1/workspace/OWOD/output/iOD/base_10/model_final.pth"
+  ROI_HEADS:
+    #    NUM_CLASSES: 21 # 0-19 Known class; 20 -> Unknown; 21 -> Background.
+    NUM_CLASSES: 21
+#    NMS_THRESH_TEST: 0.2
+#    SCORE_THRESH_TEST: 0.001
+  ROI_BOX_HEAD:
+    CLS_AGNOSTIC_BBOX_REG: True
+DATASETS:
+  TRAIN: ('voc_2007_trainval', )
+  TEST: ('voc_2007_test', )
+SOLVER:
+  STEPS: (32000, 34000)
+  MAX_ITER: 36000
+  WARMUP_ITERS: 100
+OUTPUT_DIR: "./output/iOD/10_p_10_test"
+OWOD:
+  ENABLE_THRESHOLD_AUTOLABEL_UNK: True
+  ENABLE_CLUSTERING: False
+  PREV_INTRODUCED_CLS: 15
+  CUR_INTRODUCED_CLS: 5
+  CLUSTERING:
+    START_ITER: 20000
+    UPDATE_MU_ITER: 3100
+SEED: 9999

configs/OWOD/t1.yaml

@@ -0,0 +1,16 @@
+_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

configs/OWOD/t1/t1_test.yaml

@@ -0,0 +1,18 @@
+_BASE_: "../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+  WEIGHTS: "/home/joseph/workspace/OWOD/output/t1_clustering_with_save/model_final.pth"
+  ROI_HEADS:
+    NMS_THRESH_TEST: 0.4
+TEST:
+  DETECTIONS_PER_IMAGE: 50
+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
+SOLVER:
+  STEPS: (60000, 80000)
+  MAX_ITER: 90000
+  WARMUP_ITERS: 100
+OUTPUT_DIR: "./output/temp_3"
+OWOD:
+  PREV_INTRODUCED_CLS: 0
+  CUR_INTRODUCED_CLS: 20

configs/OWOD/t1/t1_test.yaml-bak2

@@ -0,0 +1,15 @@
+_BASE_: "../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+  WEIGHTS: "/home/joseph/workspace/OWOD/output/t1_clustering_with_save/model_final.pth"
+  ROI_HEADS:
+    NMS_THRESH_TEST: 0.4
+TEST:
+  DETECTIONS_PER_IMAGE: 50
+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
+SOLVER:
+  STEPS: (60000, 80000)
+  MAX_ITER: 90000
+  WARMUP_ITERS: 100
+OUTPUT_DIR: "./output/temp_3"

configs/OWOD/t1/t1_train.yaml

@@ -0,0 +1,14 @@
+_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

configs/OWOD/t1/t1_train_fasterrcnn.yaml

@@ -0,0 +1,16 @@
+_BASE_: "../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+  WEIGHTS: "../R-50.pkl" # "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
+  # ENABLE_THRESHOLD_AUTOLABEL_UNK: False
+  ENABLE_CLUSTERING: False

configs/OWOD/t1/t1_train_fasterrcnn.yaml-bak

@@ -0,0 +1,11 @@
+_BASE_: "../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+  WEIGHTS: "../R-50.pkl" # "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"

configs/OWOD/t1/t1_train_ours.yaml

@@ -0,0 +1,16 @@
+_BASE_: "../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+  WEIGHTS: "../R-50.pkl" # "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
+  # ENABLE_THRESHOLD_AUTOLABEL_UNK: False
+  ENABLE_CLUSTERING: False

configs/OWOD/t1/t1_train_with_unk.yaml

@@ -0,0 +1,16 @@
+_BASE_: "../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
+DATASETS:
+  TRAIN: ('t1_voc_coco_2007_val_with_known_and_unk', ) # 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_train_with_known_and_unk"
+OWOD:
+  PREV_INTRODUCED_CLS: 0
+  CUR_INTRODUCED_CLS: 20
+  ENABLE_CLUSTERING: False
+  ENABLE_THRESHOLD_AUTOLABEL_UNK: False

configs/OWOD/t1/t1_val.yaml

@@ -0,0 +1,19 @@
+_BASE_: "../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+  WEIGHTS: "/home/joseph/workspace/OWOD/output/t1_clustering_with_save/model_final.pth"
+DATASETS:
+  TRAIN: ('voc_coco_2007_val', ) # t1_voc_coco_2007_train, t1_voc_coco_2007_ft
+  TEST: ('voc_coco_2007_val', )   # voc_coco_2007_test
+SOLVER:
+  STEPS: (12000, 16000)
+  MAX_ITER: 500
+  WARMUP_ITERS: 0
+OUTPUT_DIR: "./output/temp_3"
+OWOD:
+  PREV_INTRODUCED_CLS: 0
+  CUR_INTRODUCED_CLS: 20
+  COMPUTE_ENERGY: True
+  ENERGY_SAVE_PATH: 'energy'
+  SKIP_TRAINING_WHILE_EVAL: False
+  ENABLE_CLUSTERING: False
+  TEMPERATURE: 1.5

configs/OWOD/t2/t2_ft.yaml

@@ -0,0 +1,18 @@
+_BASE_: "../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+  WEIGHTS: "/home/fk1/workspace/OWOD/output/t2_ft_10/model_final.pth"
+DATASETS:
+  TRAIN: ('t2_voc_coco_2007_ft', ) # t1_voc_coco_2007_train, t1_voc_coco_2007_ft
+  TEST: ('voc_coco_2007_test', )
+SOLVER:
+  STEPS: (122000, 132000)
+  MAX_ITER: 146000
+  WARMUP_ITERS: 0
+OUTPUT_DIR: "./output/t2_ft"
+OWOD:
+  PREV_INTRODUCED_CLS: 20
+  CUR_INTRODUCED_CLS: 20
+  ENABLE_CLUSTERING: False
+  ENABLE_THRESHOLD_AUTOLABEL_UNK: False
+  CLUSTERING:
+    START_ITER: 90000

configs/OWOD/t2/t2_ft_10.yaml

@@ -0,0 +1,20 @@
+_BASE_: "../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+#  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
+#  WEIGHTS: "/home/fk1/workspace/OWOD/output/t1_ENABLE_CLUSTERING/model_final.pth"
+  WEIGHTS: "/home/fk1/workspace/OWOD/output/t2_ft_10/model_final.pth"
+DATASETS:
+  TRAIN: ('t2_voc_coco_2007_ft_10', ) # t1_voc_coco_2007_train, t1_voc_coco_2007_ft
+  TEST: ('voc_coco_2007_test', )
+SOLVER:
+  STEPS: (50000, 60000)
+  MAX_ITER: 74000
+  WARMUP_ITERS: 0
+OUTPUT_DIR: "./output/t2_ft_10"
+OWOD:
+  PREV_INTRODUCED_CLS: 20
+  CUR_INTRODUCED_CLS: 20
+  ENABLE_CLUSTERING: False
+  ENABLE_THRESHOLD_AUTOLABEL_UNK: False
+  CLUSTERING:
+    START_ITER: 18000

configs/OWOD/t2/t2_ft_20.yaml

@@ -0,0 +1,20 @@
+_BASE_: "../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+#  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
+#  WEIGHTS: "/home/fk1/workspace/OWOD/output/t1_ENABLE_CLUSTERING/model_final.pth"
+  WEIGHTS: "/home/fk1/workspace/OWOD/output/t2_ft_100/model_final.pth"
+DATASETS:
+  TRAIN: ('t2_voc_coco_2007_ft_20', ) # t1_voc_coco_2007_train, t1_voc_coco_2007_ft
+  TEST: ('voc_coco_2007_test', )
+SOLVER:
+  STEPS: (50000, 60000)
+  MAX_ITER: 74000
+  WARMUP_ITERS: 0
+OUTPUT_DIR: "./output/t2_ft_100"
+OWOD:
+  PREV_INTRODUCED_CLS: 20
+  CUR_INTRODUCED_CLS: 20
+  ENABLE_CLUSTERING: False
+  ENABLE_THRESHOLD_AUTOLABEL_UNK: False
+  CLUSTERING:
+    START_ITER: 18000

configs/OWOD/t2/t2_ft_200.yaml

@@ -0,0 +1,20 @@
+_BASE_: "../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+#  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
+#  WEIGHTS: "/home/fk1/workspace/OWOD/output/t1_ENABLE_CLUSTERING/model_final.pth"
+  WEIGHTS: "/home/fk1/workspace/OWOD/output/t2_ft_200/model_final.pth"
+DATASETS:
+  TRAIN: ('t2_voc_coco_2007_ft_200', ) # t1_voc_coco_2007_train, t1_voc_coco_2007_ft
+  TEST: ('voc_coco_2007_test', )
+SOLVER:
+  STEPS: (50000, 60000)
+  MAX_ITER: 74000
+  WARMUP_ITERS: 0
+OUTPUT_DIR: "./output/t2_ft_200"
+OWOD:
+  PREV_INTRODUCED_CLS: 20
+  CUR_INTRODUCED_CLS: 20
+  ENABLE_CLUSTERING: False
+  ENABLE_THRESHOLD_AUTOLABEL_UNK: False
+  CLUSTERING:
+    START_ITER: 18000

configs/OWOD/t2/t2_ft_400.yaml

@@ -0,0 +1,20 @@
+_BASE_: "../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+#  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
+#  WEIGHTS: "/home/fk1/workspace/OWOD/output/t1_ENABLE_CLUSTERING/model_final.pth"
+  WEIGHTS: "/home/fk1/workspace/OWOD/output/t2_ft_400/model_final.pth"
+DATASETS:
+  TRAIN: ('t2_voc_coco_2007_ft_400', ) # t1_voc_coco_2007_train, t1_voc_coco_2007_ft
+  TEST: ('voc_coco_2007_test', )
+SOLVER:
+  STEPS: (50000, 60000)
+  MAX_ITER: 74000
+  WARMUP_ITERS: 0
+OUTPUT_DIR: "./output/t2_ft_400"
+OWOD:
+  PREV_INTRODUCED_CLS: 20
+  CUR_INTRODUCED_CLS: 20
+  ENABLE_CLUSTERING: False
+  ENABLE_THRESHOLD_AUTOLABEL_UNK: False
+  CLUSTERING:
+    START_ITER: 18000

configs/OWOD/t2/t2_ft_50.yaml

@@ -0,0 +1,20 @@
+_BASE_: "../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+#  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
+#  WEIGHTS: "/home/fk1/workspace/OWOD/output/t1_ENABLE_CLUSTERING/model_final.pth"
+  WEIGHTS: "/home/fk1/workspace/OWOD/output/t2_ft_50/model_final.pth"
+DATASETS:
+  TRAIN: ('t2_voc_coco_2007_ft_50', ) # t1_voc_coco_2007_train, t1_voc_coco_2007_ft
+  TEST: ('voc_coco_2007_test', )
+SOLVER:
+  STEPS: (50000, 60000)
+  MAX_ITER: 74000
+  WARMUP_ITERS: 0
+OUTPUT_DIR: "./output/t2_ft_50"
+OWOD:
+  PREV_INTRODUCED_CLS: 20
+  CUR_INTRODUCED_CLS: 20
+  ENABLE_CLUSTERING: False
+  ENABLE_THRESHOLD_AUTOLABEL_UNK: False
+  CLUSTERING:
+    START_ITER: 18000

configs/OWOD/t2/t2_test.yaml

@@ -0,0 +1,16 @@
+_BASE_: "../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+  WEIGHTS: "/home/joseph/workspace/OWOD/output/t2_ft/model_final.pth"
+TEST:
+  DETECTIONS_PER_IMAGE: 50
+DATASETS:
+  TRAIN: ('t2_voc_coco_2007_train', ) # t1_voc_coco_2007_train, t1_voc_coco_2007_ft
+  TEST: ('voc_coco_2007_test', )
+SOLVER:
+  STEPS: (122000, 132000)
+  MAX_ITER: 142000
+  WARMUP_ITERS: 0
+OUTPUT_DIR: "./output/t2_evaluate"
+OWOD:
+  PREV_INTRODUCED_CLS: 20
+  CUR_INTRODUCED_CLS: 20

configs/OWOD/t2/t2_train.yaml

@@ -0,0 +1,20 @@
+_BASE_: "../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+  WEIGHTS: "/home/joseph/workspace/OWOD/output/t2/model_final.pth"
+DATASETS:
+  TRAIN: ('t2_voc_coco_2007_train', ) # t1_voc_coco_2007_train, t1_voc_coco_2007_ft
+  TEST: ('voc_coco_2007_test', )
+SOLVER:
+  STEPS: (122000, 132000)
+  MAX_ITER: 142000
+  WARMUP_ITERS: 0
+OUTPUT_DIR: "./output/t2"
+OWOD:
+  PREV_INTRODUCED_CLS: 20
+  CUR_INTRODUCED_CLS: 20
+  CLUSTERING:
+    ITEMS_PER_CLASS: 20
+    START_ITER: 90000 # 90000
+    UPDATE_MU_ITER: 3000
+    MOMENTUM: 0.99
+    Z_DIMENSION: 128

configs/OWOD/t2/t2_train_baseline_only_frcnn.yaml

@@ -0,0 +1,16 @@
+_BASE_: "../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+  WEIGHTS: "/home/fk1/workspace/OWOD/output/t2_baseline/model_final.pth"
+DATASETS:
+  TRAIN: ('t2_voc_coco_2007_train', )
+  TEST: ('voc_coco_2007_test', )
+SOLVER:
+  STEPS: (50000, 60000)
+  MAX_ITER: 70000
+  WARMUP_ITERS: 0
+OUTPUT_DIR: "./output/t2_baseline"
+OWOD:
+  PREV_INTRODUCED_CLS: 20
+  CUR_INTRODUCED_CLS: 20
+  ENABLE_THRESHOLD_AUTOLABEL_UNK: False
+  ENABLE_CLUSTERING: False

configs/OWOD/t2/t2_train_fasterrcnn.yaml

@@ -0,0 +1,16 @@
+_BASE_: "../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+  WEIGHTS: "/home/joseph/workspace/OWOD/output/t2/model_final.pth"
+DATASETS:
+  TRAIN: ('t2_voc_coco_2007_train', ) # t1_voc_coco_2007_train, t1_voc_coco_2007_ft
+  TEST: ('voc_coco_2007_test', )
+SOLVER:
+  STEPS: (122000, 132000)
+  MAX_ITER: 142000
+  WARMUP_ITERS: 0
+OUTPUT_DIR: "./output/t2"
+OWOD:
+  PREV_INTRODUCED_CLS: 20
+  CUR_INTRODUCED_CLS: 20
+  ENABLE_THRESHOLD_AUTOLABEL_UNK: False
+  ENABLE_CLUSTERING: False

configs/OWOD/t2/t2_train_with_unk.yaml

@@ -0,0 +1,22 @@
+_BASE_: "../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
+DATASETS:
+  TRAIN: ('t2_voc_coco_2007_val_with_known_and_unk', ) # t1_voc_coco_2007_train, t1_voc_coco_2007_ft
+  TEST: ('voc_coco_2007_test', )
+SOLVER:
+  STEPS: (122000, 132000)
+  MAX_ITER: 142000
+  WARMUP_ITERS: 0
+OUTPUT_DIR: "./output/t2_train_with_unk"
+OWOD:
+  ENABLE_CLUSTERING: False
+  ENABLE_THRESHOLD_AUTOLABEL_UNK: False
+  PREV_INTRODUCED_CLS: 20
+  CUR_INTRODUCED_CLS: 20
+  CLUSTERING:
+    ITEMS_PER_CLASS: 20
+    START_ITER: 90000
+    UPDATE_MU_ITER: 3000
+    MOMENTUM: 0.99
+    Z_DIMENSION: 128

configs/OWOD/t2/t2_val.yaml

@@ -0,0 +1,18 @@
+_BASE_: "../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+  WEIGHTS: "/home/joseph/workspace/OWOD/output/t2_ft/model_final.pth"
+DATASETS:
+  TRAIN: ('voc_coco_2007_val', ) # t1_voc_coco_2007_train, t1_voc_coco_2007_ft
+  TEST: ('voc_coco_2007_val', )
+SOLVER:
+  STEPS: (50000, 60000)
+  MAX_ITER: 500
+  WARMUP_ITERS: 0
+OUTPUT_DIR: "./output/t2_val"
+OWOD:
+  PREV_INTRODUCED_CLS: 20
+  CUR_INTRODUCED_CLS: 20
+  ENABLE_CLUSTERING: False
+  COMPUTE_ENERGY: True
+  ENERGY_SAVE_PATH: 'energy'
+  SKIP_TRAINING_WHILE_EVAL: False

configs/OWOD/t3/t3_ft.yaml

@@ -0,0 +1,18 @@
+_BASE_: "../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+  WEIGHTS: "/home/fk1/workspace/OWOD/output/t3_ft/model_final.pth"
+DATASETS:
+  TRAIN: ('t3_voc_coco_2007_ft', )
+  TEST: ('voc_coco_2007_test', )
+SOLVER:
+  STEPS: (182000, 184000)
+  MAX_ITER: 188500
+  WARMUP_ITERS: 0
+OUTPUT_DIR: "./output/t3_ft_baseline"
+OWOD:
+  PREV_INTRODUCED_CLS: 40
+  CUR_INTRODUCED_CLS: 20
+  ENABLE_CLUSTERING: False
+  ENABLE_THRESHOLD_AUTOLABEL_UNK: False
+  CLUSTERING:
+    START_ITER: 146000

configs/OWOD/t3/t3_test.yaml

@@ -0,0 +1,18 @@
+_BASE_: "../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+  WEIGHTS: "/home/joseph/workspace/OWOD/output/t3_ft/model_final.pth"
+  ROI_HEADS:
+    NMS_THRESH_TEST: 0.4
+TEST:
+  DETECTIONS_PER_IMAGE: 100
+DATASETS:
+  TRAIN: ('t3_voc_coco_2007_train', )
+  TEST: ('voc_coco_2007_test', )
+SOLVER:
+  STEPS: (182000, 184000)
+  MAX_ITER: 186000
+  WARMUP_ITERS: 0
+OUTPUT_DIR: "./output/t3_evalulate"
+OWOD:
+  PREV_INTRODUCED_CLS: 40
+  CUR_INTRODUCED_CLS: 20

configs/OWOD/t3/t3_train.yaml

@@ -0,0 +1,16 @@
+_BASE_: "../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+  WEIGHTS: "/home/fk1/workspace/OWOD/output/t3/model_final.pth"
+DATASETS:
+  TRAIN: ('t3_voc_coco_2007_train', )
+  TEST: ('voc_coco_2007_test', )
+SOLVER:
+  STEPS: (182000, 184000)
+  MAX_ITER: 186000
+  WARMUP_ITERS: 0
+OUTPUT_DIR: "./output/t3"
+OWOD:
+  PREV_INTRODUCED_CLS: 40
+  CUR_INTRODUCED_CLS: 20
+  CLUSTERING:
+    START_ITER: 146000

configs/OWOD/t3/t3_train_baseline_only_frcnn.yaml

@@ -0,0 +1,16 @@
+_BASE_: "../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+  WEIGHTS: "/home/fk1/workspace/OWOD/output/t3_baseline/model_final.pth"
+DATASETS:
+  TRAIN: ('t3_voc_coco_2007_train', )
+  TEST: ('voc_coco_2007_test', )
+SOLVER:
+  STEPS: (110000, 112000)
+  MAX_ITER: 114000
+  WARMUP_ITERS: 0
+OUTPUT_DIR: "./output/t3_baseline"
+OWOD:
+  PREV_INTRODUCED_CLS: 40
+  CUR_INTRODUCED_CLS: 20
+  ENABLE_THRESHOLD_AUTOLABEL_UNK: False
+  ENABLE_CLUSTERING: False

configs/OWOD/t3/t3_train_fasterrcnn.yaml

@@ -0,0 +1,16 @@
+_BASE_: "../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+  WEIGHTS: "/home/fk1/workspace/OWOD/output/t3/model_final.pth"
+DATASETS:
+  TRAIN: ('t3_voc_coco_2007_train', )
+  TEST: ('voc_coco_2007_test', )
+SOLVER:
+  STEPS: (182000, 184000)
+  MAX_ITER: 186000
+  WARMUP_ITERS: 0
+OUTPUT_DIR: "./output/t3"
+OWOD:
+  PREV_INTRODUCED_CLS: 40
+  CUR_INTRODUCED_CLS: 20
+  ENABLE_THRESHOLD_AUTOLABEL_UNK: False
+  ENABLE_CLUSTERING: False

configs/OWOD/t3/t3_train_with_unk.yaml

@@ -0,0 +1,18 @@
+_BASE_: "../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
+DATASETS:
+  TRAIN: ('t3_voc_coco_2007_val_with_known_and_unk', )
+  TEST: ('voc_coco_2007_test', )
+SOLVER:
+  STEPS: (182000, 184000)
+  MAX_ITER: 186000
+  WARMUP_ITERS: 0
+OUTPUT_DIR: "./output/t3_train_with_unk"
+OWOD:
+  ENABLE_CLUSTERING: False
+  ENABLE_THRESHOLD_AUTOLABEL_UNK: False
+  PREV_INTRODUCED_CLS: 40
+  CUR_INTRODUCED_CLS: 20
+  CLUSTERING:
+    START_ITER: 146000

configs/OWOD/t3/t3_val.yaml

@@ -0,0 +1,18 @@
+_BASE_: "../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+  WEIGHTS: "/home/joseph/workspace/OWOD/output/t3_ft/model_final.pth"
+DATASETS:
+  TRAIN: ('voc_coco_2007_val', ) # t1_voc_coco_2007_train, t1_voc_coco_2007_ft
+  TEST: ('voc_coco_2007_val', )
+SOLVER:
+  STEPS: (50000, 60000)
+  MAX_ITER: 500
+  WARMUP_ITERS: 0
+OUTPUT_DIR: "./output/t3_val"
+OWOD:
+  PREV_INTRODUCED_CLS: 40
+  CUR_INTRODUCED_CLS: 20
+  ENABLE_CLUSTERING: False
+  COMPUTE_ENERGY: True
+  ENERGY_SAVE_PATH: 'energy'
+  SKIP_TRAINING_WHILE_EVAL: False

configs/OWOD/t4/t4_ft.yaml

@@ -0,0 +1,18 @@
+_BASE_: "../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+  WEIGHTS: "/home/fk1/workspace/OWOD/output/t4_ft/model_final.pth"
+DATASETS:
+  TRAIN: ('t4_voc_coco_2007_ft', )
+  TEST: ('voc_coco_2007_test', )
+SOLVER:
+  STEPS: (227500, 228500)
+  MAX_ITER: 232500
+  WARMUP_ITERS: 0
+OUTPUT_DIR: "./output/t4_ft"
+OWOD:
+  PREV_INTRODUCED_CLS: 60
+  CUR_INTRODUCED_CLS: 20
+  ENABLE_CLUSTERING: False
+  ENABLE_THRESHOLD_AUTOLABEL_UNK: False
+  CLUSTERING:
+    START_ITER: 188500

configs/OWOD/t4/t4_test.yaml

@@ -0,0 +1,14 @@
+_BASE_: "../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+  WEIGHTS: "/home/joseph/workspace/OWOD/output/t4_ft/model_final.pth"
+  ROI_HEADS:
+    NMS_THRESH_TEST: 0.4
+TEST:
+  DETECTIONS_PER_IMAGE: 100
+DATASETS:
+  TRAIN: ('t3_voc_coco_2007_train', )
+  TEST: ('voc_coco_2007_test', )
+OUTPUT_DIR: "./output/t4_evalulate"
+OWOD:
+  PREV_INTRODUCED_CLS: 60
+  CUR_INTRODUCED_CLS: 20

configs/OWOD/t4/t4_train.yaml

@@ -0,0 +1,16 @@
+_BASE_: "../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+  WEIGHTS: "/home/fk1/workspace/OWOD/output/t4/model_final.pth"
+DATASETS:
+  TRAIN: ('t4_voc_coco_2007_train', )
+  TEST: ('voc_coco_2007_test', )
+SOLVER:
+  STEPS: (227500, 228500)
+  MAX_ITER: 229500
+  WARMUP_ITERS: 0
+OUTPUT_DIR: "./output/t4"
+OWOD:
+  PREV_INTRODUCED_CLS: 60
+  CUR_INTRODUCED_CLS: 20
+  CLUSTERING:
+    START_ITER: 188500

configs/OWOD/t4/t4_train_fasterrcnn.yaml

@@ -0,0 +1,16 @@
+_BASE_: "../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+  WEIGHTS: "/home/fk1/workspace/OWOD/output/t4/model_final.pth"
+DATASETS:
+  TRAIN: ('t4_voc_coco_2007_train', )
+  TEST: ('voc_coco_2007_test', )
+SOLVER:
+  STEPS: (227500, 228500)
+  MAX_ITER: 229500
+  WARMUP_ITERS: 0
+OUTPUT_DIR: "./output/t4"
+OWOD:
+  PREV_INTRODUCED_CLS: 60
+  CUR_INTRODUCED_CLS: 20
+  ENABLE_CLUSTERING: False
+  ENABLE_THRESHOLD_AUTOLABEL_UNK: False

configs/OWOD/t4/t4_train_with_all_knowns.yaml

@@ -0,0 +1,18 @@
+_BASE_: "../../Base-RCNN-C4-OWOD.yaml"
+MODEL:
+  WEIGHTS: "detectron2://ImageNetPretrained/MSRA/R-50.pkl"
+DATASETS:
+  TRAIN: ('t4_voc_coco_2007_val_with_known_and_unk', )
+  TEST: ('voc_coco_2007_test', )
+SOLVER:
+  STEPS: (155500, 156500)
+  MAX_ITER: 50000
+  WARMUP_ITERS: 0
+OUTPUT_DIR: "./output/t4_train_with_all"
+OWOD:
+  ENABLE_CLUSTERING: False
+  ENABLE_THRESHOLD_AUTOLABEL_UNK: False
+  PREV_INTRODUCED_CLS: 60
+  CUR_INTRODUCED_CLS: 20
+  CLUSTERING:
+    START_ITER: 116500