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