From 45a91e9e7d2260a02edef486917bb8a5a3bfe3c9 Mon Sep 17 00:00:00 2001
From: RE-OWOD <95522332+RE-OWOD@users.noreply.github.com>
Date: Tue, 4 Jan 2022 13:13:22 +0800
Subject: [PATCH] Add files via upload
---
configs/Base-RCNN-C4-OWOD.yaml | 40 ++++++++++++++
configs/Base-RCNN-C4-OWOD_FPN.yaml | 53 +++++++++++++++++++
configs/Base-RCNN-C4.yaml | 18 +++++++
configs/Base-RCNN-DilatedC5.yaml | 31 +++++++++++
configs/Base-RCNN-FPN.yaml | 42 +++++++++++++++
configs/Base-RetinaNet.yaml | 25 +++++++++
.../COCO-Detection/fast_rcnn_R_50_FPN_1x.yaml | 17 ++++++
.../faster_rcnn_R_101_C4_3x.yaml | 9 ++++
.../faster_rcnn_R_101_DC5_3x.yaml | 9 ++++
.../faster_rcnn_R_101_FPN_3x.yaml | 9 ++++
.../faster_rcnn_R_50_C4_1x.yaml | 6 +++
.../faster_rcnn_R_50_C4_3x.yaml | 9 ++++
.../faster_rcnn_R_50_DC5_1x.yaml | 6 +++
.../faster_rcnn_R_50_DC5_3x.yaml | 9 ++++
.../faster_rcnn_R_50_FPN_1x.yaml | 6 +++
.../faster_rcnn_R_50_FPN_3x.yaml | 9 ++++
.../faster_rcnn_X_101_32x8d_FPN_3x.yaml | 13 +++++
.../retinanet_R_101_FPN_3x.yaml | 8 +++
.../COCO-Detection/retinanet_R_50_FPN_1x.yaml | 5 ++
.../COCO-Detection/retinanet_R_50_FPN_3x.yaml | 8 +++
configs/COCO-Detection/rpn_R_50_C4_1x.yaml | 10 ++++
configs/COCO-Detection/rpn_R_50_FPN_1x.yaml | 9 ++++
configs/OWOD/iOD/10_p_10/base_10_train.yaml | 23 ++++++++
configs/OWOD/iOD/10_p_10/ft.yaml | 21 ++++++++
configs/OWOD/iOD/10_p_10/ft_with_unk.yaml | 23 ++++++++
configs/OWOD/iOD/10_p_10/next_10_train.yaml | 20 +++++++
.../next_10_train_with_unk_detection.yaml | 26 +++++++++
configs/OWOD/iOD/15_p_5/base_15_train.yaml | 23 ++++++++
configs/OWOD/iOD/15_p_5/ft_with_unk.yaml | 23 ++++++++
.../OWOD/iOD/15_p_5/next_5_train_with_ud.yaml | 26 +++++++++
configs/OWOD/iOD/19_p_1/base_19_train.yaml | 23 ++++++++
configs/OWOD/iOD/19_p_1/ft_with_unk.yaml | 23 ++++++++
.../OWOD/iOD/19_p_1/next_1_train_with_ud.yaml | 26 +++++++++
configs/OWOD/iOD/all_20_train.yaml | 23 ++++++++
configs/OWOD/iOD/test.yaml | 30 +++++++++++
configs/OWOD/t1.yaml | 16 ++++++
configs/OWOD/t1/t1_test.yaml | 18 +++++++
configs/OWOD/t1/t1_test.yaml-bak2 | 15 ++++++
configs/OWOD/t1/t1_train.yaml | 14 +++++
configs/OWOD/t1/t1_train_fasterrcnn.yaml | 16 ++++++
configs/OWOD/t1/t1_train_fasterrcnn.yaml-bak | 11 ++++
configs/OWOD/t1/t1_train_ours.yaml | 16 ++++++
configs/OWOD/t1/t1_train_with_unk.yaml | 16 ++++++
configs/OWOD/t1/t1_val.yaml | 19 +++++++
configs/OWOD/t2/t2_ft.yaml | 18 +++++++
configs/OWOD/t2/t2_ft_10.yaml | 20 +++++++
configs/OWOD/t2/t2_ft_20.yaml | 20 +++++++
configs/OWOD/t2/t2_ft_200.yaml | 20 +++++++
configs/OWOD/t2/t2_ft_400.yaml | 20 +++++++
configs/OWOD/t2/t2_ft_50.yaml | 20 +++++++
configs/OWOD/t2/t2_test.yaml | 16 ++++++
configs/OWOD/t2/t2_train.yaml | 20 +++++++
.../OWOD/t2/t2_train_baseline_only_frcnn.yaml | 16 ++++++
configs/OWOD/t2/t2_train_fasterrcnn.yaml | 16 ++++++
configs/OWOD/t2/t2_train_with_unk.yaml | 22 ++++++++
configs/OWOD/t2/t2_val.yaml | 18 +++++++
configs/OWOD/t3/t3_ft.yaml | 18 +++++++
configs/OWOD/t3/t3_test.yaml | 18 +++++++
configs/OWOD/t3/t3_train.yaml | 16 ++++++
.../OWOD/t3/t3_train_baseline_only_frcnn.yaml | 16 ++++++
configs/OWOD/t3/t3_train_fasterrcnn.yaml | 16 ++++++
configs/OWOD/t3/t3_train_with_unk.yaml | 18 +++++++
configs/OWOD/t3/t3_val.yaml | 18 +++++++
configs/OWOD/t4/t4_ft.yaml | 18 +++++++
configs/OWOD/t4/t4_test.yaml | 14 +++++
configs/OWOD/t4/t4_train.yaml | 16 ++++++
configs/OWOD/t4/t4_train_fasterrcnn.yaml | 16 ++++++
configs/OWOD/t4/t4_train_with_all_knowns.yaml | 18 +++++++
68 files changed, 1230 insertions(+)
create mode 100644 configs/Base-RCNN-C4-OWOD.yaml
create mode 100644 configs/Base-RCNN-C4-OWOD_FPN.yaml
create mode 100644 configs/Base-RCNN-C4.yaml
create mode 100644 configs/Base-RCNN-DilatedC5.yaml
create mode 100644 configs/Base-RCNN-FPN.yaml
create mode 100644 configs/Base-RetinaNet.yaml
create mode 100644 configs/COCO-Detection/fast_rcnn_R_50_FPN_1x.yaml
create mode 100644 configs/COCO-Detection/faster_rcnn_R_101_C4_3x.yaml
create mode 100644 configs/COCO-Detection/faster_rcnn_R_101_DC5_3x.yaml
create mode 100644 configs/COCO-Detection/faster_rcnn_R_101_FPN_3x.yaml
create mode 100644 configs/COCO-Detection/faster_rcnn_R_50_C4_1x.yaml
create mode 100644 configs/COCO-Detection/faster_rcnn_R_50_C4_3x.yaml
create mode 100644 configs/COCO-Detection/faster_rcnn_R_50_DC5_1x.yaml
create mode 100644 configs/COCO-Detection/faster_rcnn_R_50_DC5_3x.yaml
create mode 100644 configs/COCO-Detection/faster_rcnn_R_50_FPN_1x.yaml
create mode 100644 configs/COCO-Detection/faster_rcnn_R_50_FPN_3x.yaml
create mode 100644 configs/COCO-Detection/faster_rcnn_X_101_32x8d_FPN_3x.yaml
create mode 100644 configs/COCO-Detection/retinanet_R_101_FPN_3x.yaml
create mode 100644 configs/COCO-Detection/retinanet_R_50_FPN_1x.yaml
create mode 100644 configs/COCO-Detection/retinanet_R_50_FPN_3x.yaml
create mode 100644 configs/COCO-Detection/rpn_R_50_C4_1x.yaml
create mode 100644 configs/COCO-Detection/rpn_R_50_FPN_1x.yaml
create mode 100644 configs/OWOD/iOD/10_p_10/base_10_train.yaml
create mode 100644 configs/OWOD/iOD/10_p_10/ft.yaml
create mode 100644 configs/OWOD/iOD/10_p_10/ft_with_unk.yaml
create mode 100644 configs/OWOD/iOD/10_p_10/next_10_train.yaml
create mode 100644 configs/OWOD/iOD/10_p_10/next_10_train_with_unk_detection.yaml
create mode 100644 configs/OWOD/iOD/15_p_5/base_15_train.yaml
create mode 100644 configs/OWOD/iOD/15_p_5/ft_with_unk.yaml
create mode 100644 configs/OWOD/iOD/15_p_5/next_5_train_with_ud.yaml
create mode 100644 configs/OWOD/iOD/19_p_1/base_19_train.yaml
create mode 100644 configs/OWOD/iOD/19_p_1/ft_with_unk.yaml
create mode 100644 configs/OWOD/iOD/19_p_1/next_1_train_with_ud.yaml
create mode 100644 configs/OWOD/iOD/all_20_train.yaml
create mode 100644 configs/OWOD/iOD/test.yaml
create mode 100644 configs/OWOD/t1.yaml
create mode 100644 configs/OWOD/t1/t1_test.yaml
create mode 100644 configs/OWOD/t1/t1_test.yaml-bak2
create mode 100644 configs/OWOD/t1/t1_train.yaml
create mode 100644 configs/OWOD/t1/t1_train_fasterrcnn.yaml
create mode 100644 configs/OWOD/t1/t1_train_fasterrcnn.yaml-bak
create mode 100644 configs/OWOD/t1/t1_train_ours.yaml
create mode 100644 configs/OWOD/t1/t1_train_with_unk.yaml
create mode 100644 configs/OWOD/t1/t1_val.yaml
create mode 100644 configs/OWOD/t2/t2_ft.yaml
create mode 100644 configs/OWOD/t2/t2_ft_10.yaml
create mode 100644 configs/OWOD/t2/t2_ft_20.yaml
create mode 100644 configs/OWOD/t2/t2_ft_200.yaml
create mode 100644 configs/OWOD/t2/t2_ft_400.yaml
create mode 100644 configs/OWOD/t2/t2_ft_50.yaml
create mode 100644 configs/OWOD/t2/t2_test.yaml
create mode 100644 configs/OWOD/t2/t2_train.yaml
create mode 100644 configs/OWOD/t2/t2_train_baseline_only_frcnn.yaml
create mode 100644 configs/OWOD/t2/t2_train_fasterrcnn.yaml
create mode 100644 configs/OWOD/t2/t2_train_with_unk.yaml
create mode 100644 configs/OWOD/t2/t2_val.yaml
create mode 100644 configs/OWOD/t3/t3_ft.yaml
create mode 100644 configs/OWOD/t3/t3_test.yaml
create mode 100644 configs/OWOD/t3/t3_train.yaml
create mode 100644 configs/OWOD/t3/t3_train_baseline_only_frcnn.yaml
create mode 100644 configs/OWOD/t3/t3_train_fasterrcnn.yaml
create mode 100644 configs/OWOD/t3/t3_train_with_unk.yaml
create mode 100644 configs/OWOD/t3/t3_val.yaml
create mode 100644 configs/OWOD/t4/t4_ft.yaml
create mode 100644 configs/OWOD/t4/t4_test.yaml
create mode 100644 configs/OWOD/t4/t4_train.yaml
create mode 100644 configs/OWOD/t4/t4_train_fasterrcnn.yaml
create mode 100644 configs/OWOD/t4/t4_train_with_all_knowns.yaml
diff --git a/configs/Base-RCNN-C4-OWOD.yaml b/configs/Base-RCNN-C4-OWOD.yaml
new file mode 100644
index 0000000..395295b
--- /dev/null
+++ b/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
diff --git a/configs/Base-RCNN-C4-OWOD_FPN.yaml b/configs/Base-RCNN-C4-OWOD_FPN.yaml
new file mode 100644
index 0000000..cd89ef9
--- /dev/null
+++ b/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
diff --git a/configs/Base-RCNN-C4.yaml b/configs/Base-RCNN-C4.yaml
new file mode 100644
index 0000000..fbf34a0
--- /dev/null
+++ b/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
diff --git a/configs/Base-RCNN-DilatedC5.yaml b/configs/Base-RCNN-DilatedC5.yaml
new file mode 100644
index 0000000..c0d6d16
--- /dev/null
+++ b/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
diff --git a/configs/Base-RCNN-FPN.yaml b/configs/Base-RCNN-FPN.yaml
new file mode 100644
index 0000000..3e020f2
--- /dev/null
+++ b/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
diff --git a/configs/Base-RetinaNet.yaml b/configs/Base-RetinaNet.yaml
new file mode 100644
index 0000000..8b45b98
--- /dev/null
+++ b/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
diff --git a/configs/COCO-Detection/fast_rcnn_R_50_FPN_1x.yaml b/configs/COCO-Detection/fast_rcnn_R_50_FPN_1x.yaml
new file mode 100644
index 0000000..773ac10
--- /dev/null
+++ b/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
diff --git a/configs/COCO-Detection/faster_rcnn_R_101_C4_3x.yaml b/configs/COCO-Detection/faster_rcnn_R_101_C4_3x.yaml
new file mode 100644
index 0000000..db142cd
--- /dev/null
+++ b/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
diff --git a/configs/COCO-Detection/faster_rcnn_R_101_DC5_3x.yaml b/configs/COCO-Detection/faster_rcnn_R_101_DC5_3x.yaml
new file mode 100644
index 0000000..bceb6b3
--- /dev/null
+++ b/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
diff --git a/configs/COCO-Detection/faster_rcnn_R_101_FPN_3x.yaml b/configs/COCO-Detection/faster_rcnn_R_101_FPN_3x.yaml
new file mode 100644
index 0000000..57a098f
--- /dev/null
+++ b/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
diff --git a/configs/COCO-Detection/faster_rcnn_R_50_C4_1x.yaml b/configs/COCO-Detection/faster_rcnn_R_50_C4_1x.yaml
new file mode 100644
index 0000000..f961301
--- /dev/null
+++ b/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
diff --git a/configs/COCO-Detection/faster_rcnn_R_50_C4_3x.yaml b/configs/COCO-Detection/faster_rcnn_R_50_C4_3x.yaml
new file mode 100644
index 0000000..bc51bce
--- /dev/null
+++ b/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
diff --git a/configs/COCO-Detection/faster_rcnn_R_50_DC5_1x.yaml b/configs/COCO-Detection/faster_rcnn_R_50_DC5_1x.yaml
new file mode 100644
index 0000000..0fe96f5
--- /dev/null
+++ b/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
diff --git a/configs/COCO-Detection/faster_rcnn_R_50_DC5_3x.yaml b/configs/COCO-Detection/faster_rcnn_R_50_DC5_3x.yaml
new file mode 100644
index 0000000..33fadeb
--- /dev/null
+++ b/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
diff --git a/configs/COCO-Detection/faster_rcnn_R_50_FPN_1x.yaml b/configs/COCO-Detection/faster_rcnn_R_50_FPN_1x.yaml
new file mode 100644
index 0000000..3262019
--- /dev/null
+++ b/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
diff --git a/configs/COCO-Detection/faster_rcnn_R_50_FPN_3x.yaml b/configs/COCO-Detection/faster_rcnn_R_50_FPN_3x.yaml
new file mode 100644
index 0000000..4139518
--- /dev/null
+++ b/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
diff --git a/configs/COCO-Detection/faster_rcnn_X_101_32x8d_FPN_3x.yaml b/configs/COCO-Detection/faster_rcnn_X_101_32x8d_FPN_3x.yaml
new file mode 100644
index 0000000..9c9b5ab
--- /dev/null
+++ b/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
diff --git a/configs/COCO-Detection/retinanet_R_101_FPN_3x.yaml b/configs/COCO-Detection/retinanet_R_101_FPN_3x.yaml
new file mode 100644
index 0000000..4abb1b9
--- /dev/null
+++ b/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
diff --git a/configs/COCO-Detection/retinanet_R_50_FPN_1x.yaml b/configs/COCO-Detection/retinanet_R_50_FPN_1x.yaml
new file mode 100644
index 0000000..4a24ce3
--- /dev/null
+++ b/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
diff --git a/configs/COCO-Detection/retinanet_R_50_FPN_3x.yaml b/configs/COCO-Detection/retinanet_R_50_FPN_3x.yaml
new file mode 100644
index 0000000..3b5412d
--- /dev/null
+++ b/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
diff --git a/configs/COCO-Detection/rpn_R_50_C4_1x.yaml b/configs/COCO-Detection/rpn_R_50_C4_1x.yaml
new file mode 100644
index 0000000..e048211
--- /dev/null
+++ b/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
diff --git a/configs/COCO-Detection/rpn_R_50_FPN_1x.yaml b/configs/COCO-Detection/rpn_R_50_FPN_1x.yaml
new file mode 100644
index 0000000..dc9c952
--- /dev/null
+++ b/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
diff --git a/configs/OWOD/iOD/10_p_10/base_10_train.yaml b/configs/OWOD/iOD/10_p_10/base_10_train.yaml
new file mode 100644
index 0000000..29ed3a9
--- /dev/null
+++ b/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
\ No newline at end of file
diff --git a/configs/OWOD/iOD/10_p_10/ft.yaml b/configs/OWOD/iOD/10_p_10/ft.yaml
new file mode 100644
index 0000000..f30760d
--- /dev/null
+++ b/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
\ No newline at end of file
diff --git a/configs/OWOD/iOD/10_p_10/ft_with_unk.yaml b/configs/OWOD/iOD/10_p_10/ft_with_unk.yaml
new file mode 100644
index 0000000..f01f7ff
--- /dev/null
+++ b/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
\ No newline at end of file
diff --git a/configs/OWOD/iOD/10_p_10/next_10_train.yaml b/configs/OWOD/iOD/10_p_10/next_10_train.yaml
new file mode 100644
index 0000000..2bbe6ed
--- /dev/null
+++ b/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
\ No newline at end of file
diff --git a/configs/OWOD/iOD/10_p_10/next_10_train_with_unk_detection.yaml b/configs/OWOD/iOD/10_p_10/next_10_train_with_unk_detection.yaml
new file mode 100644
index 0000000..5a64a4f
--- /dev/null
+++ b/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
\ No newline at end of file
diff --git a/configs/OWOD/iOD/15_p_5/base_15_train.yaml b/configs/OWOD/iOD/15_p_5/base_15_train.yaml
new file mode 100644
index 0000000..933828a
--- /dev/null
+++ b/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
\ No newline at end of file
diff --git a/configs/OWOD/iOD/15_p_5/ft_with_unk.yaml b/configs/OWOD/iOD/15_p_5/ft_with_unk.yaml
new file mode 100644
index 0000000..b7169e3
--- /dev/null
+++ b/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
\ No newline at end of file
diff --git a/configs/OWOD/iOD/15_p_5/next_5_train_with_ud.yaml b/configs/OWOD/iOD/15_p_5/next_5_train_with_ud.yaml
new file mode 100644
index 0000000..ee21ceb
--- /dev/null
+++ b/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
\ No newline at end of file
diff --git a/configs/OWOD/iOD/19_p_1/base_19_train.yaml b/configs/OWOD/iOD/19_p_1/base_19_train.yaml
new file mode 100644
index 0000000..2e7b616
--- /dev/null
+++ b/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
\ No newline at end of file
diff --git a/configs/OWOD/iOD/19_p_1/ft_with_unk.yaml b/configs/OWOD/iOD/19_p_1/ft_with_unk.yaml
new file mode 100644
index 0000000..17ddf5a
--- /dev/null
+++ b/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
\ No newline at end of file
diff --git a/configs/OWOD/iOD/19_p_1/next_1_train_with_ud.yaml b/configs/OWOD/iOD/19_p_1/next_1_train_with_ud.yaml
new file mode 100644
index 0000000..7678b4b
--- /dev/null
+++ b/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
\ No newline at end of file
diff --git a/configs/OWOD/iOD/all_20_train.yaml b/configs/OWOD/iOD/all_20_train.yaml
new file mode 100644
index 0000000..fe6495d
--- /dev/null
+++ b/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
diff --git a/configs/OWOD/iOD/test.yaml b/configs/OWOD/iOD/test.yaml
new file mode 100644
index 0000000..a19ba08
--- /dev/null
+++ b/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
\ No newline at end of file
diff --git a/configs/OWOD/t1.yaml b/configs/OWOD/t1.yaml
new file mode 100644
index 0000000..ebdc928
--- /dev/null
+++ b/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
\ No newline at end of file
diff --git a/configs/OWOD/t1/t1_test.yaml b/configs/OWOD/t1/t1_test.yaml
new file mode 100644
index 0000000..b23c63a
--- /dev/null
+++ b/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
\ No newline at end of file
diff --git a/configs/OWOD/t1/t1_test.yaml-bak2 b/configs/OWOD/t1/t1_test.yaml-bak2
new file mode 100644
index 0000000..061f27e
--- /dev/null
+++ b/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"
diff --git a/configs/OWOD/t1/t1_train.yaml b/configs/OWOD/t1/t1_train.yaml
new file mode 100644
index 0000000..0cfba4f
--- /dev/null
+++ b/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
\ No newline at end of file
diff --git a/configs/OWOD/t1/t1_train_fasterrcnn.yaml b/configs/OWOD/t1/t1_train_fasterrcnn.yaml
new file mode 100644
index 0000000..005402a
--- /dev/null
+++ b/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
diff --git a/configs/OWOD/t1/t1_train_fasterrcnn.yaml-bak b/configs/OWOD/t1/t1_train_fasterrcnn.yaml-bak
new file mode 100644
index 0000000..d355dc2
--- /dev/null
+++ b/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"
diff --git a/configs/OWOD/t1/t1_train_ours.yaml b/configs/OWOD/t1/t1_train_ours.yaml
new file mode 100644
index 0000000..005402a
--- /dev/null
+++ b/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
diff --git a/configs/OWOD/t1/t1_train_with_unk.yaml b/configs/OWOD/t1/t1_train_with_unk.yaml
new file mode 100644
index 0000000..da1e281
--- /dev/null
+++ b/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
\ No newline at end of file
diff --git a/configs/OWOD/t1/t1_val.yaml b/configs/OWOD/t1/t1_val.yaml
new file mode 100644
index 0000000..cd4b4d3
--- /dev/null
+++ b/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
\ No newline at end of file
diff --git a/configs/OWOD/t2/t2_ft.yaml b/configs/OWOD/t2/t2_ft.yaml
new file mode 100644
index 0000000..c92a4ff
--- /dev/null
+++ b/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
\ No newline at end of file
diff --git a/configs/OWOD/t2/t2_ft_10.yaml b/configs/OWOD/t2/t2_ft_10.yaml
new file mode 100644
index 0000000..0d8a321
--- /dev/null
+++ b/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
\ No newline at end of file
diff --git a/configs/OWOD/t2/t2_ft_20.yaml b/configs/OWOD/t2/t2_ft_20.yaml
new file mode 100644
index 0000000..6ed6aa7
--- /dev/null
+++ b/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
\ No newline at end of file
diff --git a/configs/OWOD/t2/t2_ft_200.yaml b/configs/OWOD/t2/t2_ft_200.yaml
new file mode 100644
index 0000000..ee0b774
--- /dev/null
+++ b/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
\ No newline at end of file
diff --git a/configs/OWOD/t2/t2_ft_400.yaml b/configs/OWOD/t2/t2_ft_400.yaml
new file mode 100644
index 0000000..64cffbd
--- /dev/null
+++ b/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
\ No newline at end of file
diff --git a/configs/OWOD/t2/t2_ft_50.yaml b/configs/OWOD/t2/t2_ft_50.yaml
new file mode 100644
index 0000000..02b519b
--- /dev/null
+++ b/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
\ No newline at end of file
diff --git a/configs/OWOD/t2/t2_test.yaml b/configs/OWOD/t2/t2_test.yaml
new file mode 100644
index 0000000..911ecf8
--- /dev/null
+++ b/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
diff --git a/configs/OWOD/t2/t2_train.yaml b/configs/OWOD/t2/t2_train.yaml
new file mode 100644
index 0000000..e79dc18
--- /dev/null
+++ b/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
\ No newline at end of file
diff --git a/configs/OWOD/t2/t2_train_baseline_only_frcnn.yaml b/configs/OWOD/t2/t2_train_baseline_only_frcnn.yaml
new file mode 100644
index 0000000..3987ed0
--- /dev/null
+++ b/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
\ No newline at end of file
diff --git a/configs/OWOD/t2/t2_train_fasterrcnn.yaml b/configs/OWOD/t2/t2_train_fasterrcnn.yaml
new file mode 100644
index 0000000..f076e3c
--- /dev/null
+++ b/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
\ No newline at end of file
diff --git a/configs/OWOD/t2/t2_train_with_unk.yaml b/configs/OWOD/t2/t2_train_with_unk.yaml
new file mode 100644
index 0000000..e454684
--- /dev/null
+++ b/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
\ No newline at end of file
diff --git a/configs/OWOD/t2/t2_val.yaml b/configs/OWOD/t2/t2_val.yaml
new file mode 100644
index 0000000..73b2ef9
--- /dev/null
+++ b/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
\ No newline at end of file
diff --git a/configs/OWOD/t3/t3_ft.yaml b/configs/OWOD/t3/t3_ft.yaml
new file mode 100644
index 0000000..2e5864b
--- /dev/null
+++ b/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
\ No newline at end of file
diff --git a/configs/OWOD/t3/t3_test.yaml b/configs/OWOD/t3/t3_test.yaml
new file mode 100644
index 0000000..d279a14
--- /dev/null
+++ b/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
\ No newline at end of file
diff --git a/configs/OWOD/t3/t3_train.yaml b/configs/OWOD/t3/t3_train.yaml
new file mode 100644
index 0000000..8e8ce65
--- /dev/null
+++ b/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
\ No newline at end of file
diff --git a/configs/OWOD/t3/t3_train_baseline_only_frcnn.yaml b/configs/OWOD/t3/t3_train_baseline_only_frcnn.yaml
new file mode 100644
index 0000000..584b884
--- /dev/null
+++ b/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
\ No newline at end of file
diff --git a/configs/OWOD/t3/t3_train_fasterrcnn.yaml b/configs/OWOD/t3/t3_train_fasterrcnn.yaml
new file mode 100644
index 0000000..d0d5415
--- /dev/null
+++ b/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
\ No newline at end of file
diff --git a/configs/OWOD/t3/t3_train_with_unk.yaml b/configs/OWOD/t3/t3_train_with_unk.yaml
new file mode 100644
index 0000000..53bc794
--- /dev/null
+++ b/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
\ No newline at end of file
diff --git a/configs/OWOD/t3/t3_val.yaml b/configs/OWOD/t3/t3_val.yaml
new file mode 100644
index 0000000..4146998
--- /dev/null
+++ b/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
\ No newline at end of file
diff --git a/configs/OWOD/t4/t4_ft.yaml b/configs/OWOD/t4/t4_ft.yaml
new file mode 100644
index 0000000..e997a74
--- /dev/null
+++ b/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
\ No newline at end of file
diff --git a/configs/OWOD/t4/t4_test.yaml b/configs/OWOD/t4/t4_test.yaml
new file mode 100644
index 0000000..b43106f
--- /dev/null
+++ b/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
\ No newline at end of file
diff --git a/configs/OWOD/t4/t4_train.yaml b/configs/OWOD/t4/t4_train.yaml
new file mode 100644
index 0000000..b09b24d
--- /dev/null
+++ b/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
\ No newline at end of file
diff --git a/configs/OWOD/t4/t4_train_fasterrcnn.yaml b/configs/OWOD/t4/t4_train_fasterrcnn.yaml
new file mode 100644
index 0000000..9f952b8
--- /dev/null
+++ b/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
\ No newline at end of file
diff --git a/configs/OWOD/t4/t4_train_with_all_knowns.yaml b/configs/OWOD/t4/t4_train_with_all_knowns.yaml
new file mode 100644
index 0000000..0c90f9d
--- /dev/null
+++ b/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
\ No newline at end of file