From 69f4407267afbe5cac025d2906d9d789d170d4f9 Mon Sep 17 00:00:00 2001 From: liaoxingyu Date: Wed, 30 Sep 2020 16:05:58 +0800 Subject: [PATCH] update distill reid project (#287) Summary: refactor distill reid code and config files --- projects/DistillReID/README.md | 6 +- .../KD-bot101ibn-bot18.yml | 53 ------- .../KD-bot101ibn-bot18ibn.yml | 53 ------- .../KD-bot101ibn-bot34.yml | 53 ------- .../KD-bot101ibn-bot34ibn.yml | 53 ------- .../KD-bot101ibn-bot50.yml | 53 ------- .../KD-bot101ibn-bot50ibn.yml | 53 ------- .../KD-bot34ibn-bot18ibn.yml | 53 ------- .../KD-bot50ibn-bot18ibn.yml | 53 ------- .../KD-bot50ibn-bot34ibn.yml | 53 ------- .../bagtricks_R101-ibn.yml | 12 -- .../bagtricks_R18-ibn.yml | 14 -- .../bagtricks_R34-ibn.yml | 14 -- .../bagtricks_R50-ibn.yml | 12 -- .../KD-bot101ibn-bot18.yml | 53 ------- .../KD-bot101ibn-bot18ibn.yml | 53 ------- .../KD-bot101ibn-bot34.yml | 53 ------- .../KD-bot101ibn-bot34ibn.yml | 53 ------- .../KD-bot101ibn-bot50.yml | 53 ------- .../KD-bot101ibn-bot50ibn.yml | 53 ------- .../KD-bot34ibn-bot18ibn.yml | 53 ------- .../KD-bot50ibn-bot18ibn.yml | 53 ------- .../KD-bot50ibn-bot34ibn.yml | 53 ------- .../bagtricks_R101-ibn.yml | 12 -- .../bagtricks_R18-ibn.yml | 14 -- .../bagtricks_R34-ibn.yml | 14 -- .../bagtricks_R50-ibn.yml | 12 -- .../KD-bot101ibn-bot18.yml | 53 ------- .../KD-bot101ibn-bot18ibn.yml | 53 ------- .../KD-bot101ibn-bot34.yml | 53 ------- .../KD-bot101ibn-bot34ibn.yml | 53 ------- .../KD-bot101ibn-bot50.yml | 53 ------- .../KD-bot101ibn-bot50ibn.yml | 53 ------- .../KD-bot34ibn-bot18ibn.yml | 53 ------- .../KD-bot50ibn-bot18ibn.yml | 53 ------- .../KD-bot50ibn-bot34ibn.yml | 53 ------- .../bagtricks_R101-ibn.yml | 12 -- .../bagtricks_R18-ibn.yml | 14 -- .../bagtricks_R34-ibn.yml | 14 -- .../bagtricks_R50-ibn.yml | 12 -- .../KD-sbs101ibn-sbs18ibn.yml | 50 ------- .../KD-sbs101ibn-sbs34ibn.yml | 51 ------- .../KD-sbs101ibn-sbs50ibn.yml | 51 ------- .../KD-sbs34ibn-sbs18ibn.yml | 50 ------- .../KD-sbs50ibn-sbs18ibn.yml | 50 ------- .../KD-sbs50ibn-sbs34ibn.yml | 51 ------- .../sbs_R34-ibn.yml | 16 --- .../KD-sbs101ibn-sbs18ibn.yml | 50 ------- .../KD-sbs101ibn-sbs34ibn.yml | 51 ------- .../KD-sbs101ibn-sbs50ibn.yml | 51 ------- .../KD-sbs34ibn-sbs18ibn.yml | 50 ------- .../KD-sbs50ibn-sbs18ibn.yml | 50 ------- .../KD-sbs50ibn-sbs34ibn.yml | 51 ------- .../sbs_R101-ibn.yml | 14 -- .../sbs_R18-ibn.yml | 16 --- .../sbs_R34-ibn.yml | 16 --- .../sbs_R50-ibn.yml | 16 --- .../KD-sbs101ibn-sbs18ibn.yml | 50 ------- .../KD-sbs101ibn-sbs34ibn.yml | 51 ------- .../KD-sbs34ibn-sbs18ibn.yml | 50 ------- .../KD-sbs50ibn-sbs18ibn.yml | 50 ------- .../KD-sbs50ibn-sbs34ibn.yml | 51 ------- .../configs-sbs-ibn-msmt17/sbs_R101-ibn.yml | 14 -- .../configs-sbs-ibn-msmt17/sbs_R18-ibn.yml | 16 --- .../configs-sbs-ibn-msmt17/sbs_R34-ibn.yml | 16 --- .../configs-sbs-ibn-msmt17/sbs_R50-ibn.yml | 16 --- projects/DistillReID/configs/Base-bot-kd.yml | 30 ++++ .../Base-sbs-kd.yml} | 38 ++--- .../DukeMTMC/KD-bot101ibn-bot18ibn.yml | 20 +++ .../DukeMTMC/KD-bot101ibn-bot50ibn.yml | 20 +++ .../configs/DukeMTMC/KD-bot50ibn-bot18ibn.yml | 20 +++ .../DukeMTMC/KD-sbs101ibn-sbs18ibn.yml | 20 +++ .../DukeMTMC/KD-sbs101ibn-sbs50ibn.yml | 20 +++ .../configs/DukeMTMC/KD-sbs50ibn-sbs18ibn.yml | 20 +++ .../configs/DukeMTMC/bot101ibn.yml | 12 ++ .../DistillReID/configs/DukeMTMC/bot18ibn.yml | 13 ++ .../DistillReID/configs/DukeMTMC/bot50ibn.yml | 12 ++ .../DukeMTMC/sbs101ibn.yml} | 5 +- .../DukeMTMC/sbs18ibn.yml} | 7 +- .../DukeMTMC/sbs50ibn.yml} | 7 +- projects/DistillReID/kdreid/config.py | 12 +- projects/DistillReID/kdreid/kd_trainer.py | 41 +++--- .../backbones/shufflenetv2/network.py | 4 +- projects/DistillReID/run_bagtricks.sh | 132 ------------------ 84 files changed, 233 insertions(+), 2791 deletions(-) delete mode 100644 projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot18.yml delete mode 100644 projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot18ibn.yml delete mode 100644 projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot34.yml delete mode 100644 projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot34ibn.yml delete mode 100644 projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot50.yml delete mode 100644 projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot50ibn.yml delete mode 100644 projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot34ibn-bot18ibn.yml delete mode 100644 projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot50ibn-bot18ibn.yml delete mode 100644 projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot50ibn-bot34ibn.yml delete mode 100644 projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/bagtricks_R101-ibn.yml delete mode 100644 projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/bagtricks_R18-ibn.yml delete mode 100644 projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/bagtricks_R34-ibn.yml delete mode 100644 projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/bagtricks_R50-ibn.yml delete mode 100644 projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot18.yml delete mode 100644 projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot18ibn.yml delete mode 100644 projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot34.yml delete mode 100644 projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot34ibn.yml delete mode 100644 projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot50.yml delete mode 100644 projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot50ibn.yml delete mode 100644 projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot34ibn-bot18ibn.yml delete mode 100644 projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot50ibn-bot18ibn.yml delete mode 100644 projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot50ibn-bot34ibn.yml delete mode 100644 projects/DistillReID/configs-bagtricks-ibn-market1501/bagtricks_R101-ibn.yml delete mode 100644 projects/DistillReID/configs-bagtricks-ibn-market1501/bagtricks_R18-ibn.yml delete mode 100644 projects/DistillReID/configs-bagtricks-ibn-market1501/bagtricks_R34-ibn.yml delete mode 100644 projects/DistillReID/configs-bagtricks-ibn-market1501/bagtricks_R50-ibn.yml delete mode 100644 projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot18.yml delete mode 100644 projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot18ibn.yml delete mode 100644 projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot34.yml delete mode 100644 projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot34ibn.yml delete mode 100644 projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot50.yml delete mode 100644 projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot50ibn.yml delete mode 100644 projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot34ibn-bot18ibn.yml delete mode 100644 projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot50ibn-bot18ibn.yml delete mode 100644 projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot50ibn-bot34ibn.yml delete mode 100644 projects/DistillReID/configs-bagtricks-ibn-msmt17/bagtricks_R101-ibn.yml delete mode 100644 projects/DistillReID/configs-bagtricks-ibn-msmt17/bagtricks_R18-ibn.yml delete mode 100644 projects/DistillReID/configs-bagtricks-ibn-msmt17/bagtricks_R34-ibn.yml delete mode 100644 projects/DistillReID/configs-bagtricks-ibn-msmt17/bagtricks_R50-ibn.yml delete mode 100644 projects/DistillReID/configs-sbs-ibn-dukemtmcreid/KD-sbs101ibn-sbs18ibn.yml delete mode 100644 projects/DistillReID/configs-sbs-ibn-dukemtmcreid/KD-sbs101ibn-sbs34ibn.yml delete mode 100644 projects/DistillReID/configs-sbs-ibn-dukemtmcreid/KD-sbs101ibn-sbs50ibn.yml delete mode 100644 projects/DistillReID/configs-sbs-ibn-dukemtmcreid/KD-sbs34ibn-sbs18ibn.yml delete mode 100644 projects/DistillReID/configs-sbs-ibn-dukemtmcreid/KD-sbs50ibn-sbs18ibn.yml delete mode 100644 projects/DistillReID/configs-sbs-ibn-dukemtmcreid/KD-sbs50ibn-sbs34ibn.yml delete mode 100644 projects/DistillReID/configs-sbs-ibn-dukemtmcreid/sbs_R34-ibn.yml delete mode 100644 projects/DistillReID/configs-sbs-ibn-market1501/KD-sbs101ibn-sbs18ibn.yml delete mode 100644 projects/DistillReID/configs-sbs-ibn-market1501/KD-sbs101ibn-sbs34ibn.yml delete mode 100644 projects/DistillReID/configs-sbs-ibn-market1501/KD-sbs101ibn-sbs50ibn.yml delete mode 100644 projects/DistillReID/configs-sbs-ibn-market1501/KD-sbs34ibn-sbs18ibn.yml delete mode 100644 projects/DistillReID/configs-sbs-ibn-market1501/KD-sbs50ibn-sbs18ibn.yml delete mode 100644 projects/DistillReID/configs-sbs-ibn-market1501/KD-sbs50ibn-sbs34ibn.yml delete mode 100644 projects/DistillReID/configs-sbs-ibn-market1501/sbs_R101-ibn.yml delete mode 100644 projects/DistillReID/configs-sbs-ibn-market1501/sbs_R18-ibn.yml delete mode 100644 projects/DistillReID/configs-sbs-ibn-market1501/sbs_R34-ibn.yml delete mode 100644 projects/DistillReID/configs-sbs-ibn-market1501/sbs_R50-ibn.yml delete mode 100644 projects/DistillReID/configs-sbs-ibn-msmt17/KD-sbs101ibn-sbs18ibn.yml delete mode 100644 projects/DistillReID/configs-sbs-ibn-msmt17/KD-sbs101ibn-sbs34ibn.yml delete mode 100644 projects/DistillReID/configs-sbs-ibn-msmt17/KD-sbs34ibn-sbs18ibn.yml delete mode 100644 projects/DistillReID/configs-sbs-ibn-msmt17/KD-sbs50ibn-sbs18ibn.yml delete mode 100644 projects/DistillReID/configs-sbs-ibn-msmt17/KD-sbs50ibn-sbs34ibn.yml delete mode 100644 projects/DistillReID/configs-sbs-ibn-msmt17/sbs_R101-ibn.yml delete mode 100644 projects/DistillReID/configs-sbs-ibn-msmt17/sbs_R18-ibn.yml delete mode 100644 projects/DistillReID/configs-sbs-ibn-msmt17/sbs_R34-ibn.yml delete mode 100644 projects/DistillReID/configs-sbs-ibn-msmt17/sbs_R50-ibn.yml create mode 100644 projects/DistillReID/configs/Base-bot-kd.yml rename projects/DistillReID/{configs-sbs-ibn-msmt17/KD-sbs101ibn-sbs50ibn.yml => configs/Base-sbs-kd.yml} (50%) create mode 100644 projects/DistillReID/configs/DukeMTMC/KD-bot101ibn-bot18ibn.yml create mode 100644 projects/DistillReID/configs/DukeMTMC/KD-bot101ibn-bot50ibn.yml create mode 100644 projects/DistillReID/configs/DukeMTMC/KD-bot50ibn-bot18ibn.yml create mode 100644 projects/DistillReID/configs/DukeMTMC/KD-sbs101ibn-sbs18ibn.yml create mode 100644 projects/DistillReID/configs/DukeMTMC/KD-sbs101ibn-sbs50ibn.yml create mode 100644 projects/DistillReID/configs/DukeMTMC/KD-sbs50ibn-sbs18ibn.yml create mode 100644 projects/DistillReID/configs/DukeMTMC/bot101ibn.yml create mode 100644 projects/DistillReID/configs/DukeMTMC/bot18ibn.yml create mode 100644 projects/DistillReID/configs/DukeMTMC/bot50ibn.yml rename projects/DistillReID/{configs-sbs-ibn-dukemtmcreid/sbs_R101-ibn.yml => configs/DukeMTMC/sbs101ibn.yml} (66%) rename projects/DistillReID/{configs-sbs-ibn-dukemtmcreid/sbs_R18-ibn.yml => configs/DukeMTMC/sbs18ibn.yml} (58%) rename projects/DistillReID/{configs-sbs-ibn-dukemtmcreid/sbs_R50-ibn.yml => configs/DukeMTMC/sbs50ibn.yml} (57%) delete mode 100644 projects/DistillReID/run_bagtricks.sh diff --git a/projects/DistillReID/README.md b/projects/DistillReID/README.md index ef0fa54..6ba91ee 100644 --- a/projects/DistillReID/README.md +++ b/projects/DistillReID/README.md @@ -15,9 +15,9 @@ This project provides a training script of small model # a demo on DukeMTMC-reID dataset # please see more in ./configs # train BagTricksIBN50 as teacher model -CUDA_VISIBLE_DEVICES=$CUDA python ./tools/train_net.py --config-file ./projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/bagtricks_R50-ibn.yml MODEL.DEVICE "cuda:0" +python3 projects/DistillReID/train_net.py --config-file projects/DistillReID/configs/DukeMTMC/bot50ibn.yml # train BagTricksIBN18 as student model -CUDA_VISIBLE_DEVICES=$CUDA python ./projects/DistillReID/train_net.py --kd --config-file ./projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot50ibn-bot18ibn.yml MODEL.DEVICE "cuda:0" +python3 projects/DistillReID/train_net.py --kd --config-file projects/DistillReID/configs/DukeMTMC/KD-bot50ibn-bot18ibn.yml ``` ## Experimental Results and Trained Models @@ -42,6 +42,6 @@ All the experiments are conducted with a P40 GPU and ## Contact -This project is conducted by [Guan'an Wang](https://wangguanan.github.io/) (guan.wang0706@gmail) and [Liao Xingyu](https://github.com/L1aoXingyu). +This project is conducted by [Guan'an Wang](https://wangguanan.github.io/) (guan.wang0706@gmail) and [Xingyu Liao](https://github.com/L1aoXingyu). diff --git a/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot18.yml b/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot18.yml deleted file mode 100644 index 3cbc5fb..0000000 --- a/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot18.yml +++ /dev/null @@ -1,53 +0,0 @@ -_BASE_: "../../../configs/Base-bagtricks.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "101x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "18x" - LAST_STRIDE: 1 - WITH_IBN: False - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 512 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/dukemtmc/bagtricks_R101-ibn/model_final.pth" -DATASETS: - NAMES: ("DukeMTMC",) - TESTS: ("DukeMTMC",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/dukemtmc/bot101ibn-kd-bot18" \ No newline at end of file diff --git a/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot18ibn.yml b/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot18ibn.yml deleted file mode 100644 index b269ad0..0000000 --- a/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot18ibn.yml +++ /dev/null @@ -1,53 +0,0 @@ -_BASE_: "../../../configs/Base-bagtricks.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "101x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "18x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 512 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/dukemtmc/bagtricks_R101-ibn/model_final.pth" -DATASETS: - NAMES: ("DukeMTMC",) - TESTS: ("DukeMTMC",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/dukemtmc/bot101ibn-kd-bot18ibn" \ No newline at end of file diff --git a/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot34.yml b/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot34.yml deleted file mode 100644 index 4195a11..0000000 --- a/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot34.yml +++ /dev/null @@ -1,53 +0,0 @@ -_BASE_: "../../../configs/Base-bagtricks.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "101x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "34x" - LAST_STRIDE: 1 - WITH_IBN: False - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 512 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/dukemtmc/bagtricks_R101-ibn/model_final.pth" -DATASETS: - NAMES: ("DukeMTMC",) - TESTS: ("DukeMTMC",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/dukemtmc/bot101ibn-kd-bot34" \ No newline at end of file diff --git a/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot34ibn.yml b/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot34ibn.yml deleted file mode 100644 index daa26c7..0000000 --- a/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot34ibn.yml +++ /dev/null @@ -1,53 +0,0 @@ -_BASE_: "../../../configs/Base-bagtricks.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "101x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "34x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 512 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/dukemtmc/bagtricks_R101-ibn/model_final.pth" -DATASETS: - NAMES: ("DukeMTMC",) - TESTS: ("DukeMTMC",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/dukemtmc/bot101ibn-kd-bot34ibn" \ No newline at end of file diff --git a/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot50.yml b/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot50.yml deleted file mode 100644 index 2f93e9a..0000000 --- a/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot50.yml +++ /dev/null @@ -1,53 +0,0 @@ -_BASE_: "../../../configs/Base-bagtricks.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "101x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "50x" - LAST_STRIDE: 1 - WITH_IBN: False - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/dukemtmc/bagtricks_R101-ibn/model_final.pth" -DATASETS: - NAMES: ("DukeMTMC",) - TESTS: ("DukeMTMC",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/dukemtmc/bot101ibn-kd-bot50" \ No newline at end of file diff --git a/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot50ibn.yml b/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot50ibn.yml deleted file mode 100644 index 557e078..0000000 --- a/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot50ibn.yml +++ /dev/null @@ -1,53 +0,0 @@ -_BASE_: "../../../configs/Base-bagtricks.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "101x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "50x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/dukemtmc/bagtricks_R101-ibn/model_final.pth" -DATASETS: - NAMES: ("DukeMTMC",) - TESTS: ("DukeMTMC",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/dukemtmc/bot101ibn-kd-bot50ibn" \ No newline at end of file diff --git a/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot34ibn-bot18ibn.yml b/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot34ibn-bot18ibn.yml deleted file mode 100644 index 565a965..0000000 --- a/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot34ibn-bot18ibn.yml +++ /dev/null @@ -1,53 +0,0 @@ -_BASE_: "../../../configs/Base-bagtricks.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "34x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 512 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "18x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 512 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/dukemtmc/bagtricks_R34-ibn/model_final.pth" -DATASETS: - NAMES: ("DukeMTMC",) - TESTS: ("DukeMTMC",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/dukemtmc/bot34ibn-kd-bot18ibn" \ No newline at end of file diff --git a/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot50ibn-bot18ibn.yml b/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot50ibn-bot18ibn.yml deleted file mode 100644 index e65f165..0000000 --- a/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot50ibn-bot18ibn.yml +++ /dev/null @@ -1,53 +0,0 @@ -_BASE_: "../../../configs/Base-bagtricks.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "50x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "18x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 512 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/dukemtmc/bagtricks_R50-ibn/model_final.pth" -DATASETS: - NAMES: ("DukeMTMC",) - TESTS: ("DukeMTMC",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/dukemtmc/bot50ibn-kd-bot18ibn" \ No newline at end of file diff --git a/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot50ibn-bot34ibn.yml b/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot50ibn-bot34ibn.yml deleted file mode 100644 index 17d2e8b..0000000 --- a/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot50ibn-bot34ibn.yml +++ /dev/null @@ -1,53 +0,0 @@ -_BASE_: "../../../configs/Base-bagtricks.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "50x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "34x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 512 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/dukemtmc/bagtricks_R50-ibn/model_final.pth" -DATASETS: - NAMES: ("DukeMTMC",) - TESTS: ("DukeMTMC",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/dukemtmc/bot50ibn-kd-bot34ibn" \ No newline at end of file diff --git a/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/bagtricks_R101-ibn.yml b/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/bagtricks_R101-ibn.yml deleted file mode 100644 index 9eaeef0..0000000 --- a/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/bagtricks_R101-ibn.yml +++ /dev/null @@ -1,12 +0,0 @@ -_BASE_: "../../../configs/Base-bagtricks.yml" - -MODEL: - BACKBONE: - DEPTH: '101x' - WITH_IBN: True - -DATASETS: - NAMES: ("DukeMTMC",) - TESTS: ("DukeMTMC",) - -OUTPUT_DIR: "logs/dukemtmc/bagtricks_R101-ibn" diff --git a/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/bagtricks_R18-ibn.yml b/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/bagtricks_R18-ibn.yml deleted file mode 100644 index fdac078..0000000 --- a/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/bagtricks_R18-ibn.yml +++ /dev/null @@ -1,14 +0,0 @@ -_BASE_: "../../../configs/Base-bagtricks.yml" - -MODEL: - BACKBONE: - DEPTH: "18x" - WITH_IBN: True - HEADS: - IN_FEAT: 512 - -DATASETS: - NAMES: ("DukeMTMC",) - TESTS: ("DukeMTMC",) - -OUTPUT_DIR: "logs/dukemtmc/bagtricks_R18-ibn" diff --git a/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/bagtricks_R34-ibn.yml b/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/bagtricks_R34-ibn.yml deleted file mode 100644 index 04500f9..0000000 --- a/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/bagtricks_R34-ibn.yml +++ /dev/null @@ -1,14 +0,0 @@ -_BASE_: "../../../configs/Base-bagtricks.yml" - -MODEL: - BACKBONE: - DEPTH: "34x" - WITH_IBN: True - HEADS: - IN_FEAT: 512 - -DATASETS: - NAMES: ("DukeMTMC",) - TESTS: ("DukeMTMC",) - -OUTPUT_DIR: "logs/dukemtmc/bagtricks_R34-ibn" diff --git a/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/bagtricks_R50-ibn.yml b/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/bagtricks_R50-ibn.yml deleted file mode 100644 index 2adb188..0000000 --- a/projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/bagtricks_R50-ibn.yml +++ /dev/null @@ -1,12 +0,0 @@ -_BASE_: "../../../configs/Base-bagtricks.yml" - -MODEL: - BACKBONE: - DEPTH: "50x" - WITH_IBN: True - -DATASETS: - NAMES: ("DukeMTMC",) - TESTS: ("DukeMTMC",) - -OUTPUT_DIR: "logs/dukemtmc/bagtricks_R50-ibn" diff --git a/projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot18.yml b/projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot18.yml deleted file mode 100644 index 1de80a4..0000000 --- a/projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot18.yml +++ /dev/null @@ -1,53 +0,0 @@ -_BASE_: "../../../configs/Base-bagtricks.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "101x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "18x" - LAST_STRIDE: 1 - WITH_IBN: False - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 512 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/market1501/bagtricks_R101-ibn/model_final.pth" -DATASETS: - NAMES: ("Market1501",) - TESTS: ("Market1501",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/market1501/bot101ibn-kd-bot18" \ No newline at end of file diff --git a/projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot18ibn.yml b/projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot18ibn.yml deleted file mode 100644 index ddd38e4..0000000 --- a/projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot18ibn.yml +++ /dev/null @@ -1,53 +0,0 @@ -_BASE_: "../../../configs/Base-bagtricks.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "101x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "18x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 512 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/market1501/bagtricks_R101-ibn/model_final.pth" -DATASETS: - NAMES: ("Market1501",) - TESTS: ("Market1501",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/market1501/bot101ibn-kd-bot18ibn" \ No newline at end of file diff --git a/projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot34.yml b/projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot34.yml deleted file mode 100644 index 93ffaff..0000000 --- a/projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot34.yml +++ /dev/null @@ -1,53 +0,0 @@ -_BASE_: "../../../configs/Base-bagtricks.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "101x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "34x" - LAST_STRIDE: 1 - WITH_IBN: False - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 512 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/market1501/bagtricks_R101-ibn/model_final.pth" -DATASETS: - NAMES: ("Market1501",) - TESTS: ("Market1501",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/market1501/bot101ibn-kd-bot34" \ No newline at end of file diff --git a/projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot34ibn.yml b/projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot34ibn.yml deleted file mode 100644 index c288b7c..0000000 --- a/projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot34ibn.yml +++ /dev/null @@ -1,53 +0,0 @@ -_BASE_: "../../../configs/Base-bagtricks.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "101x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "34x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 512 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/market1501/bagtricks_R101-ibn/model_final.pth" -DATASETS: - NAMES: ("Market1501",) - TESTS: ("Market1501",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/market1501/bot101ibn-kd-bot34ibn" \ No newline at end of file diff --git a/projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot50.yml b/projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot50.yml deleted file mode 100644 index 6246652..0000000 --- a/projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot50.yml +++ /dev/null @@ -1,53 +0,0 @@ -_BASE_: "../../../configs/Base-bagtricks.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "101x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "50x" - LAST_STRIDE: 1 - WITH_IBN: False - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/market1501/bagtricks_R101-ibn/model_final.pth" -DATASETS: - NAMES: ("Market1501",) - TESTS: ("Market1501",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/market1501/bot101ibn-kd-bot50" \ No newline at end of file diff --git a/projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot50ibn.yml b/projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot50ibn.yml deleted file mode 100644 index 26b1e9a..0000000 --- a/projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot50ibn.yml +++ /dev/null @@ -1,53 +0,0 @@ -_BASE_: "../../../configs/Base-bagtricks.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "101x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "50x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/market1501/bagtricks_R101-ibn/model_final.pth" -DATASETS: - NAMES: ("Market1501",) - TESTS: ("Market1501",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/market1501/bot101ibn-kd-bot50ibn" \ No newline at end of file diff --git a/projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot34ibn-bot18ibn.yml b/projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot34ibn-bot18ibn.yml deleted file mode 100644 index 6580e60..0000000 --- a/projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot34ibn-bot18ibn.yml +++ /dev/null @@ -1,53 +0,0 @@ -_BASE_: "../../../configs/Base-bagtricks.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "34x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 512 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "18x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 512 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/market1501/bagtricks_R34-ibn/model_final.pth" -DATASETS: - NAMES: ("Market1501",) - TESTS: ("Market1501",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/market1501/bot34ibn-kd-bot18ibn" \ No newline at end of file diff --git a/projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot50ibn-bot18ibn.yml b/projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot50ibn-bot18ibn.yml deleted file mode 100644 index 64f084b..0000000 --- a/projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot50ibn-bot18ibn.yml +++ /dev/null @@ -1,53 +0,0 @@ -_BASE_: "../../../configs/Base-bagtricks.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "50x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "18x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 512 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/market1501/bagtricks_R50-ibn/model_final.pth" -DATASETS: - NAMES: ("Market1501",) - TESTS: ("Market1501",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/market1501/bot50ibn-kd-bot18ibn" \ No newline at end of file diff --git a/projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot50ibn-bot34ibn.yml b/projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot50ibn-bot34ibn.yml deleted file mode 100644 index 0f2f3be..0000000 --- a/projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot50ibn-bot34ibn.yml +++ /dev/null @@ -1,53 +0,0 @@ -_BASE_: "../../../configs/Base-bagtricks.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "50x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "34x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 512 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/market1501/bagtricks_R50-ibn/model_final.pth" -DATASETS: - NAMES: ("Market1501",) - TESTS: ("Market1501",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/market1501/bot50ibn-kd-bot34ibn" \ No newline at end of file diff --git a/projects/DistillReID/configs-bagtricks-ibn-market1501/bagtricks_R101-ibn.yml b/projects/DistillReID/configs-bagtricks-ibn-market1501/bagtricks_R101-ibn.yml deleted file mode 100644 index d824632..0000000 --- a/projects/DistillReID/configs-bagtricks-ibn-market1501/bagtricks_R101-ibn.yml +++ /dev/null @@ -1,12 +0,0 @@ -_BASE_: "../../../configs/Base-bagtricks.yml" - -MODEL: - BACKBONE: - DEPTH: '101x' - WITH_IBN: True - -DATASETS: - NAMES: ("Market1501",) - TESTS: ("Market1501",) - -OUTPUT_DIR: "logs/market1501/bagtricks_R101-ibn" diff --git a/projects/DistillReID/configs-bagtricks-ibn-market1501/bagtricks_R18-ibn.yml b/projects/DistillReID/configs-bagtricks-ibn-market1501/bagtricks_R18-ibn.yml deleted file mode 100644 index caf5f41..0000000 --- a/projects/DistillReID/configs-bagtricks-ibn-market1501/bagtricks_R18-ibn.yml +++ /dev/null @@ -1,14 +0,0 @@ -_BASE_: "../../../configs/Base-bagtricks.yml" - -MODEL: - BACKBONE: - DEPTH: "18x" - WITH_IBN: True - HEADS: - IN_FEAT: 512 - -DATASETS: - NAMES: ("Market1501",) - TESTS: ("Market1501",) - -OUTPUT_DIR: "logs/market1501/bagtricks_R18-ibn" diff --git a/projects/DistillReID/configs-bagtricks-ibn-market1501/bagtricks_R34-ibn.yml b/projects/DistillReID/configs-bagtricks-ibn-market1501/bagtricks_R34-ibn.yml deleted file mode 100644 index fea7a96..0000000 --- a/projects/DistillReID/configs-bagtricks-ibn-market1501/bagtricks_R34-ibn.yml +++ /dev/null @@ -1,14 +0,0 @@ -_BASE_: "../../../configs/Base-bagtricks.yml" - -MODEL: - BACKBONE: - DEPTH: "34x" - WITH_IBN: True - HEADS: - IN_FEAT: 512 - -DATASETS: - NAMES: ("Market1501",) - TESTS: ("Market1501",) - -OUTPUT_DIR: "logs/market1501/bagtricks_R34-ibn" diff --git a/projects/DistillReID/configs-bagtricks-ibn-market1501/bagtricks_R50-ibn.yml b/projects/DistillReID/configs-bagtricks-ibn-market1501/bagtricks_R50-ibn.yml deleted file mode 100644 index 733743d..0000000 --- a/projects/DistillReID/configs-bagtricks-ibn-market1501/bagtricks_R50-ibn.yml +++ /dev/null @@ -1,12 +0,0 @@ -_BASE_: "../../../configs/Base-bagtricks.yml" - -MODEL: - BACKBONE: - DEPTH: "50x" - WITH_IBN: True - -DATASETS: - NAMES: ("Market1501",) - TESTS: ("Market1501",) - -OUTPUT_DIR: "logs/market1501/bagtricks_R50-ibn" diff --git a/projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot18.yml b/projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot18.yml deleted file mode 100644 index 733ba90..0000000 --- a/projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot18.yml +++ /dev/null @@ -1,53 +0,0 @@ -_BASE_: "../../../configs/Base-bagtricks.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "101x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "18x" - LAST_STRIDE: 1 - WITH_IBN: False - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 512 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/market1501/bagtricks_R101-ibn/model_final.pth" -DATASETS: - NAMES: ("MSMT17",) - TESTS: ("MSMT17",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/msmt17/bot101ibn-kd-bot18" \ No newline at end of file diff --git a/projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot18ibn.yml b/projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot18ibn.yml deleted file mode 100644 index e80fb07..0000000 --- a/projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot18ibn.yml +++ /dev/null @@ -1,53 +0,0 @@ -_BASE_: "../../../configs/Base-bagtricks.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "101x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "18x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 512 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/market1501/bagtricks_R101-ibn/model_final.pth" -DATASETS: - NAMES: ("MSMT17",) - TESTS: ("MSMT17",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/msmt17/bot101ibn-kd-bot18ibn" \ No newline at end of file diff --git a/projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot34.yml b/projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot34.yml deleted file mode 100644 index 07fa025..0000000 --- a/projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot34.yml +++ /dev/null @@ -1,53 +0,0 @@ -_BASE_: "../../../configs/Base-bagtricks.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "101x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "34x" - LAST_STRIDE: 1 - WITH_IBN: False - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 512 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/market1501/bagtricks_R101-ibn/model_final.pth" -DATASETS: - NAMES: ("MSMT17",) - TESTS: ("MSMT17",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/msmt17/bot101ibn-kd-bot34" \ No newline at end of file diff --git a/projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot34ibn.yml b/projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot34ibn.yml deleted file mode 100644 index 1088b6b..0000000 --- a/projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot34ibn.yml +++ /dev/null @@ -1,53 +0,0 @@ -_BASE_: "../../../configs/Base-bagtricks.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "101x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "34x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 512 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/market1501/bagtricks_R101-ibn/model_final.pth" -DATASETS: - NAMES: ("MSMT17",) - TESTS: ("MSMT17",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/msmt17/bot101ibn-kd-bot34ibn" \ No newline at end of file diff --git a/projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot50.yml b/projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot50.yml deleted file mode 100644 index 65a1dd9..0000000 --- a/projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot50.yml +++ /dev/null @@ -1,53 +0,0 @@ -_BASE_: "../../../configs/Base-bagtricks.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "101x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "50x" - LAST_STRIDE: 1 - WITH_IBN: False - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/market1501/bagtricks_R101-ibn/model_final.pth" -DATASETS: - NAMES: ("MSMT17",) - TESTS: ("MSMT17",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/msmt17/bot101ibn-kd-bot50" \ No newline at end of file diff --git a/projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot50ibn.yml b/projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot50ibn.yml deleted file mode 100644 index 18afdb5..0000000 --- a/projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot50ibn.yml +++ /dev/null @@ -1,53 +0,0 @@ -_BASE_: "../../../configs/Base-bagtricks.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "101x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "50x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/market1501/bagtricks_R101-ibn/model_final.pth" -DATASETS: - NAMES: ("MSMT17",) - TESTS: ("MSMT17",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/msmt17/bot101ibn-kd-bot50ibn" \ No newline at end of file diff --git a/projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot34ibn-bot18ibn.yml b/projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot34ibn-bot18ibn.yml deleted file mode 100644 index d86f881..0000000 --- a/projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot34ibn-bot18ibn.yml +++ /dev/null @@ -1,53 +0,0 @@ -_BASE_: "../../../configs/Base-bagtricks.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "34x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 512 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "18x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 512 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/market1501/bagtricks_R34-ibn/model_final.pth" -DATASETS: - NAMES: ("MSMT17",) - TESTS: ("MSMT17",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/msmt17/bot34ibn-kd-bot18ibn" \ No newline at end of file diff --git a/projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot50ibn-bot18ibn.yml b/projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot50ibn-bot18ibn.yml deleted file mode 100644 index a9738ec..0000000 --- a/projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot50ibn-bot18ibn.yml +++ /dev/null @@ -1,53 +0,0 @@ -_BASE_: "../../../configs/Base-bagtricks.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "50x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "18x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 512 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/market1501/bagtricks_R50-ibn/model_final.pth" -DATASETS: - NAMES: ("MSMT17",) - TESTS: ("MSMT17",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/msmt17/bot50ibn-kd-bot18ibn" \ No newline at end of file diff --git a/projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot50ibn-bot34ibn.yml b/projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot50ibn-bot34ibn.yml deleted file mode 100644 index 65202af..0000000 --- a/projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot50ibn-bot34ibn.yml +++ /dev/null @@ -1,53 +0,0 @@ -_BASE_: "../../../configs/Base-bagtricks.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "50x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "34x" - LAST_STRIDE: 1 - WITH_IBN: True - PRETRAIN: True - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 512 - NORM: "BN" - POOL_LAYER: "avgpool" - NECK_FEAT: "before" - CLS_LAYER: "linear" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/market1501/bagtricks_R50-ibn/model_final.pth" -DATASETS: - NAMES: ("MSMT17",) - TESTS: ("MSMT17",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/msmt17/bot50ibn-kd-bot34ibn" \ No newline at end of file diff --git a/projects/DistillReID/configs-bagtricks-ibn-msmt17/bagtricks_R101-ibn.yml b/projects/DistillReID/configs-bagtricks-ibn-msmt17/bagtricks_R101-ibn.yml deleted file mode 100644 index ac9f6a4..0000000 --- a/projects/DistillReID/configs-bagtricks-ibn-msmt17/bagtricks_R101-ibn.yml +++ /dev/null @@ -1,12 +0,0 @@ -_BASE_: "../../../configs/Base-bagtricks.yml" - -MODEL: - BACKBONE: - DEPTH: '101x' - WITH_IBN: True - -DATASETS: - NAMES: ("MSMT17",) - TESTS: ("MSMT17",) - -OUTPUT_DIR: "logs/msmt17/bagtricks_R101-ibn" diff --git a/projects/DistillReID/configs-bagtricks-ibn-msmt17/bagtricks_R18-ibn.yml b/projects/DistillReID/configs-bagtricks-ibn-msmt17/bagtricks_R18-ibn.yml deleted file mode 100644 index 01b6ffb..0000000 --- a/projects/DistillReID/configs-bagtricks-ibn-msmt17/bagtricks_R18-ibn.yml +++ /dev/null @@ -1,14 +0,0 @@ -_BASE_: "../../../configs/Base-bagtricks.yml" - -MODEL: - BACKBONE: - DEPTH: "18x" - WITH_IBN: True - HEADS: - IN_FEAT: 512 - -DATASETS: - NAMES: ("MSMT17",) - TESTS: ("MSMT17",) - -OUTPUT_DIR: "logs/msmt17/bagtricks_R18-ibn" diff --git a/projects/DistillReID/configs-bagtricks-ibn-msmt17/bagtricks_R34-ibn.yml b/projects/DistillReID/configs-bagtricks-ibn-msmt17/bagtricks_R34-ibn.yml deleted file mode 100644 index 2775aeb..0000000 --- a/projects/DistillReID/configs-bagtricks-ibn-msmt17/bagtricks_R34-ibn.yml +++ /dev/null @@ -1,14 +0,0 @@ -_BASE_: "../../../configs/Base-bagtricks.yml" - -MODEL: - BACKBONE: - DEPTH: "34x" - WITH_IBN: True - HEADS: - IN_FEAT: 512 - -DATASETS: - NAMES: ("MSMT17",) - TESTS: ("MSMT17",) - -OUTPUT_DIR: "logs/msmt17/bagtricks_R34-ibn" diff --git a/projects/DistillReID/configs-bagtricks-ibn-msmt17/bagtricks_R50-ibn.yml b/projects/DistillReID/configs-bagtricks-ibn-msmt17/bagtricks_R50-ibn.yml deleted file mode 100644 index 3edd27f..0000000 --- a/projects/DistillReID/configs-bagtricks-ibn-msmt17/bagtricks_R50-ibn.yml +++ /dev/null @@ -1,12 +0,0 @@ -_BASE_: "../../../configs/Base-bagtricks.yml" - -MODEL: - BACKBONE: - DEPTH: "50x" - WITH_IBN: True - -DATASETS: - NAMES: ("MSMT17",) - TESTS: ("MSMT17",) - -OUTPUT_DIR: "logs/msmt17/bagtricks_R50-ibn" diff --git a/projects/DistillReID/configs-sbs-ibn-dukemtmcreid/KD-sbs101ibn-sbs18ibn.yml b/projects/DistillReID/configs-sbs-ibn-dukemtmcreid/KD-sbs101ibn-sbs18ibn.yml deleted file mode 100644 index 84a278a..0000000 --- a/projects/DistillReID/configs-sbs-ibn-dukemtmcreid/KD-sbs101ibn-sbs18ibn.yml +++ /dev/null @@ -1,50 +0,0 @@ -_BASE_: "../../../configs/Base-Strongerbaseline.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "101x" - LAST_STRIDE: 1 - WITH_NL: True - WITH_IBN: True - PRETRAIN: False - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - NECK_FEAT: "after" - POOL_LAYER: "gempool" - CLS_LAYER: "circle" - SCALE: 64 - MARGIN: 0.35 - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - WITH_NL: False - DEPTH: "18x" - WITH_IBN: True - HEADS: - IN_FEAT: 512 - CLS_LAYER: "circleSoftmax" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/dukemtmcreid/sbs_R101-ibn/model_final.pth" - - -DATASETS: - NAMES: ("DukeMTMC",) - TESTS: ("DukeMTMC",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/dukemtmcreid/sbs101ibn-kd-sbs18ibn" diff --git a/projects/DistillReID/configs-sbs-ibn-dukemtmcreid/KD-sbs101ibn-sbs34ibn.yml b/projects/DistillReID/configs-sbs-ibn-dukemtmcreid/KD-sbs101ibn-sbs34ibn.yml deleted file mode 100644 index 4f0c131..0000000 --- a/projects/DistillReID/configs-sbs-ibn-dukemtmcreid/KD-sbs101ibn-sbs34ibn.yml +++ /dev/null @@ -1,51 +0,0 @@ -_BASE_: "../../../configs/Base-Strongerbaseline.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "101x" - LAST_STRIDE: 1 - WITH_NL: True - WITH_IBN: True - PRETRAIN: False - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - NECK_FEAT: "after" - POOL_LAYER: "gempool" - CLS_LAYER: "circle" - SCALE: 64 - MARGIN: 0.35 - - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - WITH_NL: False - DEPTH: "34x" - WITH_IBN: True - HEADS: - IN_FEAT: 512 - CLS_LAYER: "circleSoftmax" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/dukemtmcreid/sbs_R101-ibn/model_final.pth" - - -DATASETS: - NAMES: ("DukeMTMC",) - TESTS: ("DukeMTMC",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/dukemtmcreid/sbs101ibn-kd-sbs34ibn" diff --git a/projects/DistillReID/configs-sbs-ibn-dukemtmcreid/KD-sbs101ibn-sbs50ibn.yml b/projects/DistillReID/configs-sbs-ibn-dukemtmcreid/KD-sbs101ibn-sbs50ibn.yml deleted file mode 100644 index 8fc9e19..0000000 --- a/projects/DistillReID/configs-sbs-ibn-dukemtmcreid/KD-sbs101ibn-sbs50ibn.yml +++ /dev/null @@ -1,51 +0,0 @@ -_BASE_: "../../../configs/Base-Strongerbaseline.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "101x" - LAST_STRIDE: 1 - WITH_NL: True - WITH_IBN: True - PRETRAIN: False - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - NECK_FEAT: "after" - POOL_LAYER: "gempool" - CLS_LAYER: "circle" - SCALE: 64 - MARGIN: 0.35 - - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - WITH_NL: False - DEPTH: "50x" - WITH_IBN: True - HEADS: - IN_FEAT: 2048 - CLS_LAYER: "circleSoftmax" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/dukemtmcreid/sbs_R101-ibn/model_final.pth" - - -DATASETS: - NAMES: ("DukeMTMC",) - TESTS: ("DukeMTMC",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/dukemtmcreid/sbs101ibn-kd-sbs50ibn" \ No newline at end of file diff --git a/projects/DistillReID/configs-sbs-ibn-dukemtmcreid/KD-sbs34ibn-sbs18ibn.yml b/projects/DistillReID/configs-sbs-ibn-dukemtmcreid/KD-sbs34ibn-sbs18ibn.yml deleted file mode 100644 index a1dbb73..0000000 --- a/projects/DistillReID/configs-sbs-ibn-dukemtmcreid/KD-sbs34ibn-sbs18ibn.yml +++ /dev/null @@ -1,50 +0,0 @@ -_BASE_: "../../../configs/Base-Strongerbaseline.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "34x" - LAST_STRIDE: 1 - WITH_NL: True - WITH_IBN: True - PRETRAIN: False - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 512 - NORM: "BN" - NECK_FEAT: "after" - POOL_LAYER: "gempool" - CLS_LAYER: "circle" - SCALE: 64 - MARGIN: 0.35 - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - WITH_NL: False - DEPTH: "18x" - WITH_IBN: True - HEADS: - IN_FEAT: 512 - CLS_LAYER: "circleSoftmax" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/dukemtmcreid/sbs_R34-ibn/model_final.pth" - - -DATASETS: - NAMES: ("DukeMTMC",) - TESTS: ("DukeMTMC",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/dukemtmcreid/sbs34ibn-kd-sbs18ibn" diff --git a/projects/DistillReID/configs-sbs-ibn-dukemtmcreid/KD-sbs50ibn-sbs18ibn.yml b/projects/DistillReID/configs-sbs-ibn-dukemtmcreid/KD-sbs50ibn-sbs18ibn.yml deleted file mode 100644 index 06a264d..0000000 --- a/projects/DistillReID/configs-sbs-ibn-dukemtmcreid/KD-sbs50ibn-sbs18ibn.yml +++ /dev/null @@ -1,50 +0,0 @@ -_BASE_: "../../../configs/Base-Strongerbaseline.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "50x" - LAST_STRIDE: 1 - WITH_NL: True - WITH_IBN: True - PRETRAIN: False - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - NECK_FEAT: "after" - POOL_LAYER: "gempool" - CLS_LAYER: "circle" - SCALE: 64 - MARGIN: 0.35 - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - WITH_NL: False - DEPTH: "18x" - WITH_IBN: True - HEADS: - IN_FEAT: 512 - CLS_LAYER: "circleSoftmax" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/dukemtmcreid/sbs_R50-ibn/model_final.pth" - - -DATASETS: - NAMES: ("DukeMTMC",) - TESTS: ("DukeMTMC",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/dukemtmcreid/sbs50ibn-kd-sbs18ibn" diff --git a/projects/DistillReID/configs-sbs-ibn-dukemtmcreid/KD-sbs50ibn-sbs34ibn.yml b/projects/DistillReID/configs-sbs-ibn-dukemtmcreid/KD-sbs50ibn-sbs34ibn.yml deleted file mode 100644 index f75f362..0000000 --- a/projects/DistillReID/configs-sbs-ibn-dukemtmcreid/KD-sbs50ibn-sbs34ibn.yml +++ /dev/null @@ -1,51 +0,0 @@ -_BASE_: "../../../configs/Base-Strongerbaseline.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "50x" - LAST_STRIDE: 1 - WITH_NL: True - WITH_IBN: True - PRETRAIN: False - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - NECK_FEAT: "after" - POOL_LAYER: "gempool" - CLS_LAYER: "circle" - SCALE: 64 - MARGIN: 0.35 - - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - WITH_NL: False - DEPTH: "34x" - WITH_IBN: True - HEADS: - IN_FEAT: 512 - CLS_LAYER: "circleSoftmax" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/dukemtmcreid/sbs_R50-ibn/model_final.pth" - - -DATASETS: - NAMES: ("DukeMTMC",) - TESTS: ("DukeMTMC",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/dukemtmcreid/sbs50ibn-kd-sbs34ibn" diff --git a/projects/DistillReID/configs-sbs-ibn-dukemtmcreid/sbs_R34-ibn.yml b/projects/DistillReID/configs-sbs-ibn-dukemtmcreid/sbs_R34-ibn.yml deleted file mode 100644 index bc7f124..0000000 --- a/projects/DistillReID/configs-sbs-ibn-dukemtmcreid/sbs_R34-ibn.yml +++ /dev/null @@ -1,16 +0,0 @@ -_BASE_: "../../../configs/Base-Strongerbaseline.yml" - -MODEL: - BACKBONE: - WITH_NL: False - DEPTH: "34x" - WITH_IBN: True - HEADS: - IN_FEAT: 512 - CLS_LAYER: "circleSoftmax" - -DATASETS: - NAMES: ("DukeMTMC",) - TESTS: ("DukeMTMC",) - -OUTPUT_DIR: "logs/dukemtmcreid/sbs_R34-ibn" diff --git a/projects/DistillReID/configs-sbs-ibn-market1501/KD-sbs101ibn-sbs18ibn.yml b/projects/DistillReID/configs-sbs-ibn-market1501/KD-sbs101ibn-sbs18ibn.yml deleted file mode 100644 index b3517e0..0000000 --- a/projects/DistillReID/configs-sbs-ibn-market1501/KD-sbs101ibn-sbs18ibn.yml +++ /dev/null @@ -1,50 +0,0 @@ -_BASE_: "../../../configs/Base-Strongerbaseline.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "101x" - LAST_STRIDE: 1 - WITH_NL: True - WITH_IBN: True - PRETRAIN: False - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - NECK_FEAT: "after" - POOL_LAYER: "gempool" - CLS_LAYER: "circle" - SCALE: 64 - MARGIN: 0.35 - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - WITH_NL: False - DEPTH: "18x" - WITH_IBN: True - HEADS: - IN_FEAT: 512 - CLS_LAYER: "circleSoftmax" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/market1501/sbs_R101-ibn/model_final.pth" - - -DATASETS: - NAMES: ("Market1501",) - TESTS: ("Market1501",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/market1501/sbs101ibn-kd-sbs18ibn" diff --git a/projects/DistillReID/configs-sbs-ibn-market1501/KD-sbs101ibn-sbs34ibn.yml b/projects/DistillReID/configs-sbs-ibn-market1501/KD-sbs101ibn-sbs34ibn.yml deleted file mode 100644 index 5b8a504..0000000 --- a/projects/DistillReID/configs-sbs-ibn-market1501/KD-sbs101ibn-sbs34ibn.yml +++ /dev/null @@ -1,51 +0,0 @@ -_BASE_: "../../../configs/Base-Strongerbaseline.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "101x" - LAST_STRIDE: 1 - WITH_NL: True - WITH_IBN: True - PRETRAIN: False - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - NECK_FEAT: "after" - POOL_LAYER: "gempool" - CLS_LAYER: "circle" - SCALE: 64 - MARGIN: 0.35 - - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - WITH_NL: False - DEPTH: "34x" - WITH_IBN: True - HEADS: - IN_FEAT: 512 - CLS_LAYER: "circleSoftmax" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/market1501/sbs_R101-ibn/model_final.pth" - - -DATASETS: - NAMES: ("Market1501",) - TESTS: ("Market1501",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/market1501/sbs101ibn-kd-sbs34ibn" diff --git a/projects/DistillReID/configs-sbs-ibn-market1501/KD-sbs101ibn-sbs50ibn.yml b/projects/DistillReID/configs-sbs-ibn-market1501/KD-sbs101ibn-sbs50ibn.yml deleted file mode 100644 index bf1c434..0000000 --- a/projects/DistillReID/configs-sbs-ibn-market1501/KD-sbs101ibn-sbs50ibn.yml +++ /dev/null @@ -1,51 +0,0 @@ -_BASE_: "../../../configs/Base-Strongerbaseline.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "101x" - LAST_STRIDE: 1 - WITH_NL: True - WITH_IBN: True - PRETRAIN: False - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - NECK_FEAT: "after" - POOL_LAYER: "gempool" - CLS_LAYER: "circle" - SCALE: 64 - MARGIN: 0.35 - - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - WITH_NL: False - DEPTH: "50x" - WITH_IBN: True - HEADS: - IN_FEAT: 2048 - CLS_LAYER: "circleSoftmax" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/market1501/sbs_R101-ibn/model_final.pth" - - -DATASETS: - NAMES: ("Market1501",) - TESTS: ("Market1501",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/market1501/sbs101ibn-kd-sbs50ibn" \ No newline at end of file diff --git a/projects/DistillReID/configs-sbs-ibn-market1501/KD-sbs34ibn-sbs18ibn.yml b/projects/DistillReID/configs-sbs-ibn-market1501/KD-sbs34ibn-sbs18ibn.yml deleted file mode 100644 index 242059d..0000000 --- a/projects/DistillReID/configs-sbs-ibn-market1501/KD-sbs34ibn-sbs18ibn.yml +++ /dev/null @@ -1,50 +0,0 @@ -_BASE_: "../../../configs/Base-Strongerbaseline.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "34x" - LAST_STRIDE: 1 - WITH_NL: True - WITH_IBN: True - PRETRAIN: False - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 512 - NORM: "BN" - NECK_FEAT: "after" - POOL_LAYER: "gempool" - CLS_LAYER: "circle" - SCALE: 64 - MARGIN: 0.35 - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - WITH_NL: False - DEPTH: "18x" - WITH_IBN: True - HEADS: - IN_FEAT: 512 - CLS_LAYER: "circleSoftmax" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/market1501/sbs_R34-ibn/model_final.pth" - - -DATASETS: - NAMES: ("Market1501",) - TESTS: ("Market1501",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/market1501/sbs34ibn-kd-sbs18ibn" diff --git a/projects/DistillReID/configs-sbs-ibn-market1501/KD-sbs50ibn-sbs18ibn.yml b/projects/DistillReID/configs-sbs-ibn-market1501/KD-sbs50ibn-sbs18ibn.yml deleted file mode 100644 index 53d3fbe..0000000 --- a/projects/DistillReID/configs-sbs-ibn-market1501/KD-sbs50ibn-sbs18ibn.yml +++ /dev/null @@ -1,50 +0,0 @@ -_BASE_: "../../../configs/Base-Strongerbaseline.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "50x" - LAST_STRIDE: 1 - WITH_NL: True - WITH_IBN: True - PRETRAIN: False - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - NECK_FEAT: "after" - POOL_LAYER: "gempool" - CLS_LAYER: "circle" - SCALE: 64 - MARGIN: 0.35 - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - WITH_NL: False - DEPTH: "18x" - WITH_IBN: True - HEADS: - IN_FEAT: 512 - CLS_LAYER: "circleSoftmax" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/market1501/sbs_R50-ibn/model_final.pth" - - -DATASETS: - NAMES: ("Market1501",) - TESTS: ("Market1501",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/market1501/sbs50ibn-kd-sbs18ibn" diff --git a/projects/DistillReID/configs-sbs-ibn-market1501/KD-sbs50ibn-sbs34ibn.yml b/projects/DistillReID/configs-sbs-ibn-market1501/KD-sbs50ibn-sbs34ibn.yml deleted file mode 100644 index 9cbb68d..0000000 --- a/projects/DistillReID/configs-sbs-ibn-market1501/KD-sbs50ibn-sbs34ibn.yml +++ /dev/null @@ -1,51 +0,0 @@ -_BASE_: "../../../configs/Base-Strongerbaseline.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "50x" - LAST_STRIDE: 1 - WITH_NL: True - WITH_IBN: True - PRETRAIN: False - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - NECK_FEAT: "after" - POOL_LAYER: "gempool" - CLS_LAYER: "circle" - SCALE: 64 - MARGIN: 0.35 - - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - WITH_NL: False - DEPTH: "34x" - WITH_IBN: True - HEADS: - IN_FEAT: 512 - CLS_LAYER: "circleSoftmax" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/market1501/sbs_R50-ibn/model_final.pth" - - -DATASETS: - NAMES: ("Market1501",) - TESTS: ("Market1501",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/market1501/sbs50ibn-kd-sbs34ibn" diff --git a/projects/DistillReID/configs-sbs-ibn-market1501/sbs_R101-ibn.yml b/projects/DistillReID/configs-sbs-ibn-market1501/sbs_R101-ibn.yml deleted file mode 100644 index 62eb7ea..0000000 --- a/projects/DistillReID/configs-sbs-ibn-market1501/sbs_R101-ibn.yml +++ /dev/null @@ -1,14 +0,0 @@ -_BASE_: "../../../configs/Base-Strongerbaseline.yml" - -MODEL: - BACKBONE: - DEPTH: "101x" - WITH_IBN: True - HEADS: - CLS_LAYER: "circleSoftmax" - -DATASETS: - NAMES: ("Market1501",) - TESTS: ("Market1501",) - -OUTPUT_DIR: "logs/market1501/sbs_R101-ibn" diff --git a/projects/DistillReID/configs-sbs-ibn-market1501/sbs_R18-ibn.yml b/projects/DistillReID/configs-sbs-ibn-market1501/sbs_R18-ibn.yml deleted file mode 100644 index 2f439d2..0000000 --- a/projects/DistillReID/configs-sbs-ibn-market1501/sbs_R18-ibn.yml +++ /dev/null @@ -1,16 +0,0 @@ -_BASE_: "../../../configs/Base-Strongerbaseline.yml" - -MODEL: - BACKBONE: - WITH_NL: False - DEPTH: "18x" - WITH_IBN: True - HEADS: - IN_FEAT: 512 - CLS_LAYER: "circleSoftmax" - -DATASETS: - NAMES: ("Market1501",) - TESTS: ("Market1501",) - -OUTPUT_DIR: "logs/market1501/sbs_R18-ibn" diff --git a/projects/DistillReID/configs-sbs-ibn-market1501/sbs_R34-ibn.yml b/projects/DistillReID/configs-sbs-ibn-market1501/sbs_R34-ibn.yml deleted file mode 100644 index b00a5b0..0000000 --- a/projects/DistillReID/configs-sbs-ibn-market1501/sbs_R34-ibn.yml +++ /dev/null @@ -1,16 +0,0 @@ -_BASE_: "../../../configs/Base-Strongerbaseline.yml" - -MODEL: - BACKBONE: - WITH_NL: False - DEPTH: "34x" - WITH_IBN: True - HEADS: - IN_FEAT: 512 - CLS_LAYER: "circleSoftmax" - -DATASETS: - NAMES: ("Market1501",) - TESTS: ("Market1501",) - -OUTPUT_DIR: "logs/market1501/sbs_R34-ibn" diff --git a/projects/DistillReID/configs-sbs-ibn-market1501/sbs_R50-ibn.yml b/projects/DistillReID/configs-sbs-ibn-market1501/sbs_R50-ibn.yml deleted file mode 100644 index 6354e48..0000000 --- a/projects/DistillReID/configs-sbs-ibn-market1501/sbs_R50-ibn.yml +++ /dev/null @@ -1,16 +0,0 @@ -_BASE_: "../../../configs/Base-Strongerbaseline.yml" - -MODEL: - BACKBONE: - WITH_NL: False - DEPTH: "50x" - WITH_IBN: True - HEADS: - IN_FEAT: 2048 - CLS_LAYER: "circleSoftmax" - -DATASETS: - NAMES: ("Market1501",) - TESTS: ("Market1501",) - -OUTPUT_DIR: "logs/market1501/sbs_R50-ibn" diff --git a/projects/DistillReID/configs-sbs-ibn-msmt17/KD-sbs101ibn-sbs18ibn.yml b/projects/DistillReID/configs-sbs-ibn-msmt17/KD-sbs101ibn-sbs18ibn.yml deleted file mode 100644 index df2bd3d..0000000 --- a/projects/DistillReID/configs-sbs-ibn-msmt17/KD-sbs101ibn-sbs18ibn.yml +++ /dev/null @@ -1,50 +0,0 @@ -_BASE_: "../../../configs/Base-Strongerbaseline.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "101x" - LAST_STRIDE: 1 - WITH_NL: True - WITH_IBN: True - PRETRAIN: False - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - NECK_FEAT: "after" - POOL_LAYER: "gempool" - CLS_LAYER: "circle" - SCALE: 64 - MARGIN: 0.35 - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - WITH_NL: False - DEPTH: "18x" - WITH_IBN: True - HEADS: - IN_FEAT: 512 - CLS_LAYER: "circleSoftmax" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/dukemtmcreid/sbs_R101-ibn/model_final.pth" - - -DATASETS: - NAMES: ("MSMT17",) - TESTS: ("MSMT17",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/dukemtmcreid/sbs101ibn-kd-sbs18ibn" diff --git a/projects/DistillReID/configs-sbs-ibn-msmt17/KD-sbs101ibn-sbs34ibn.yml b/projects/DistillReID/configs-sbs-ibn-msmt17/KD-sbs101ibn-sbs34ibn.yml deleted file mode 100644 index 717047d..0000000 --- a/projects/DistillReID/configs-sbs-ibn-msmt17/KD-sbs101ibn-sbs34ibn.yml +++ /dev/null @@ -1,51 +0,0 @@ -_BASE_: "../../../configs/Base-Strongerbaseline.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "101x" - LAST_STRIDE: 1 - WITH_NL: True - WITH_IBN: True - PRETRAIN: False - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - NECK_FEAT: "after" - POOL_LAYER: "gempool" - CLS_LAYER: "circle" - SCALE: 64 - MARGIN: 0.35 - - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - WITH_NL: False - DEPTH: "34x" - WITH_IBN: True - HEADS: - IN_FEAT: 512 - CLS_LAYER: "circleSoftmax" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/dukemtmcreid/sbs_R101-ibn/model_final.pth" - - -DATASETS: - NAMES: ("MSMT17",) - TESTS: ("MSMT17",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/dukemtmcreid/sbs101ibn-kd-sbs34ibn" diff --git a/projects/DistillReID/configs-sbs-ibn-msmt17/KD-sbs34ibn-sbs18ibn.yml b/projects/DistillReID/configs-sbs-ibn-msmt17/KD-sbs34ibn-sbs18ibn.yml deleted file mode 100644 index f5db322..0000000 --- a/projects/DistillReID/configs-sbs-ibn-msmt17/KD-sbs34ibn-sbs18ibn.yml +++ /dev/null @@ -1,50 +0,0 @@ -_BASE_: "../../../configs/Base-Strongerbaseline.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "34x" - LAST_STRIDE: 1 - WITH_NL: True - WITH_IBN: True - PRETRAIN: False - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 512 - NORM: "BN" - NECK_FEAT: "after" - POOL_LAYER: "gempool" - CLS_LAYER: "circle" - SCALE: 64 - MARGIN: 0.35 - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - WITH_NL: False - DEPTH: "18x" - WITH_IBN: True - HEADS: - IN_FEAT: 512 - CLS_LAYER: "circleSoftmax" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/dukemtmcreid/sbs_R34-ibn/model_final.pth" - - -DATASETS: - NAMES: ("MSMT17",) - TESTS: ("MSMT17",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/dukemtmcreid/sbs34ibn-kd-sbs18ibn" diff --git a/projects/DistillReID/configs-sbs-ibn-msmt17/KD-sbs50ibn-sbs18ibn.yml b/projects/DistillReID/configs-sbs-ibn-msmt17/KD-sbs50ibn-sbs18ibn.yml deleted file mode 100644 index c935831..0000000 --- a/projects/DistillReID/configs-sbs-ibn-msmt17/KD-sbs50ibn-sbs18ibn.yml +++ /dev/null @@ -1,50 +0,0 @@ -_BASE_: "../../../configs/Base-Strongerbaseline.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "50x" - LAST_STRIDE: 1 - WITH_NL: True - WITH_IBN: True - PRETRAIN: False - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - NECK_FEAT: "after" - POOL_LAYER: "gempool" - CLS_LAYER: "circle" - SCALE: 64 - MARGIN: 0.35 - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - WITH_NL: False - DEPTH: "18x" - WITH_IBN: True - HEADS: - IN_FEAT: 512 - CLS_LAYER: "circleSoftmax" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/dukemtmcreid/sbs_R50-ibn/model_final.pth" - - -DATASETS: - NAMES: ("MSMT17",) - TESTS: ("MSMT17",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/dukemtmcreid/sbs50ibn-kd-sbs18ibn" diff --git a/projects/DistillReID/configs-sbs-ibn-msmt17/KD-sbs50ibn-sbs34ibn.yml b/projects/DistillReID/configs-sbs-ibn-msmt17/KD-sbs50ibn-sbs34ibn.yml deleted file mode 100644 index 16a7e06..0000000 --- a/projects/DistillReID/configs-sbs-ibn-msmt17/KD-sbs50ibn-sbs34ibn.yml +++ /dev/null @@ -1,51 +0,0 @@ -_BASE_: "../../../configs/Base-Strongerbaseline.yml" - -SOLVER: - IMS_PER_BATCH: 64 - -MODEL_TEACHER: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - NAME: "build_resnet_backbone" - NORM: "BN" - DEPTH: "50x" - LAST_STRIDE: 1 - WITH_NL: True - WITH_IBN: True - PRETRAIN: False - - HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 - NORM: "BN" - NECK_FEAT: "after" - POOL_LAYER: "gempool" - CLS_LAYER: "circle" - SCALE: 64 - MARGIN: 0.35 - - -MODEL: - META_ARCHITECTURE: "Baseline" - - BACKBONE: - WITH_NL: False - DEPTH: "34x" - WITH_IBN: True - HEADS: - IN_FEAT: 512 - CLS_LAYER: "circleSoftmax" - - STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/dukemtmcreid/sbs_R50-ibn/model_final.pth" - - -DATASETS: - NAMES: ("MSMT17",) - TESTS: ("MSMT17",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/dukemtmcreid/sbs50ibn-kd-sbs34ibn" diff --git a/projects/DistillReID/configs-sbs-ibn-msmt17/sbs_R101-ibn.yml b/projects/DistillReID/configs-sbs-ibn-msmt17/sbs_R101-ibn.yml deleted file mode 100644 index f0d7c91..0000000 --- a/projects/DistillReID/configs-sbs-ibn-msmt17/sbs_R101-ibn.yml +++ /dev/null @@ -1,14 +0,0 @@ -_BASE_: "../../../configs/Base-Strongerbaseline.yml" - -MODEL: - BACKBONE: - DEPTH: "101x" - WITH_IBN: True - HEADS: - CLS_LAYER: "circleSoftmax" - -DATASETS: - NAMES: ("MSMT17",) - TESTS: ("MSMT17",) - -OUTPUT_DIR: "logs/dukemtmcreid/sbs_R101-ibn" diff --git a/projects/DistillReID/configs-sbs-ibn-msmt17/sbs_R18-ibn.yml b/projects/DistillReID/configs-sbs-ibn-msmt17/sbs_R18-ibn.yml deleted file mode 100644 index 8446c1f..0000000 --- a/projects/DistillReID/configs-sbs-ibn-msmt17/sbs_R18-ibn.yml +++ /dev/null @@ -1,16 +0,0 @@ -_BASE_: "../../../configs/Base-Strongerbaseline.yml" - -MODEL: - BACKBONE: - WITH_NL: False - DEPTH: "18x" - WITH_IBN: True - HEADS: - IN_FEAT: 512 - CLS_LAYER: "circleSoftmax" - -DATASETS: - NAMES: ("MSMT17",) - TESTS: ("MSMT17",) - -OUTPUT_DIR: "logs/dukemtmcreid/sbs_R18-ibn" diff --git a/projects/DistillReID/configs-sbs-ibn-msmt17/sbs_R34-ibn.yml b/projects/DistillReID/configs-sbs-ibn-msmt17/sbs_R34-ibn.yml deleted file mode 100644 index 22ad5cd..0000000 --- a/projects/DistillReID/configs-sbs-ibn-msmt17/sbs_R34-ibn.yml +++ /dev/null @@ -1,16 +0,0 @@ -_BASE_: "../../../configs/Base-Strongerbaseline.yml" - -MODEL: - BACKBONE: - WITH_NL: False - DEPTH: "34x" - WITH_IBN: True - HEADS: - IN_FEAT: 512 - CLS_LAYER: "circleSoftmax" - -DATASETS: - NAMES: ("MSMT17",) - TESTS: ("MSMT17",) - -OUTPUT_DIR: "logs/dukemtmcreid/sbs_R34-ibn" diff --git a/projects/DistillReID/configs-sbs-ibn-msmt17/sbs_R50-ibn.yml b/projects/DistillReID/configs-sbs-ibn-msmt17/sbs_R50-ibn.yml deleted file mode 100644 index d8f5895..0000000 --- a/projects/DistillReID/configs-sbs-ibn-msmt17/sbs_R50-ibn.yml +++ /dev/null @@ -1,16 +0,0 @@ -_BASE_: "../../../configs/Base-Strongerbaseline.yml" - -MODEL: - BACKBONE: - WITH_NL: False - DEPTH: "50x" - WITH_IBN: True - HEADS: - IN_FEAT: 2048 - CLS_LAYER: "circleSoftmax" - -DATASETS: - NAMES: ("MSMT17",) - TESTS: ("MSMT17",) - -OUTPUT_DIR: "logs/dukemtmcreid/sbs_R50-ibn" diff --git a/projects/DistillReID/configs/Base-bot-kd.yml b/projects/DistillReID/configs/Base-bot-kd.yml new file mode 100644 index 0000000..b02dd2e --- /dev/null +++ b/projects/DistillReID/configs/Base-bot-kd.yml @@ -0,0 +1,30 @@ +_BASE_: "../../../configs/Base-bagtricks.yml" + +MODEL_TEACHER: + META_ARCHITECTURE: "Baseline" + + BACKBONE: + NAME: "build_resnet_backbone" + NORM: "BN" + DEPTH: "101x" + FEAT_DIM: 2048 + LAST_STRIDE: 1 + WITH_IBN: True + PRETRAIN: True + + HEADS: + NAME: "EmbeddingHead" + NORM: "BN" + POOL_LAYER: "avgpool" + NECK_FEAT: "before" + CLS_LAYER: "linear" + +MODEL: + BACKBONE: + NAME: "build_resnet_backbone" + DEPTH: "50x" + FEAT_DIM: 2048 + WITH_IBN: True + + STUDENT_WEIGHTS: "" + TEACHER_WEIGHTS: "logs/dukemtmc/bagtricks_R34-ibn/model_final.pth" \ No newline at end of file diff --git a/projects/DistillReID/configs-sbs-ibn-msmt17/KD-sbs101ibn-sbs50ibn.yml b/projects/DistillReID/configs/Base-sbs-kd.yml similarity index 50% rename from projects/DistillReID/configs-sbs-ibn-msmt17/KD-sbs101ibn-sbs50ibn.yml rename to projects/DistillReID/configs/Base-sbs-kd.yml index ddf80ef..ba4cd48 100644 --- a/projects/DistillReID/configs-sbs-ibn-msmt17/KD-sbs101ibn-sbs50ibn.yml +++ b/projects/DistillReID/configs/Base-sbs-kd.yml @@ -1,8 +1,5 @@ _BASE_: "../../../configs/Base-Strongerbaseline.yml" -SOLVER: - IMS_PER_BATCH: 64 - MODEL_TEACHER: META_ARCHITECTURE: "Baseline" @@ -10,42 +7,31 @@ MODEL_TEACHER: NAME: "build_resnet_backbone" NORM: "BN" DEPTH: "101x" + FEAT_DIM: 2048 LAST_STRIDE: 1 - WITH_NL: True + WITH_NL: False WITH_IBN: True - PRETRAIN: False + PRETRAIN: True HEADS: - NAME: "BNneckHead" - IN_FEAT: 2048 + NAME: "EmbeddingHead" NORM: "BN" NECK_FEAT: "after" - POOL_LAYER: "gempool" - CLS_LAYER: "circle" + POOL_LAYER: "gempoolP" + CLS_LAYER: "circleSoftmax" SCALE: 64 MARGIN: 0.35 - MODEL: - META_ARCHITECTURE: "Baseline" - BACKBONE: - WITH_NL: False + NAME: "build_resnet_backbone" DEPTH: "50x" + FEAT_DIM: 2048 WITH_IBN: True - HEADS: - IN_FEAT: 2048 - CLS_LAYER: "circleSoftmax" STUDENT_WEIGHTS: "" - TEACHER_WEIGHTS: "./logs/dukemtmcreid/sbs_R101-ibn/model_final.pth" + TEACHER_WEIGHTS: "logs/dukemtmc/bagtricks_R34-ibn/model_final.pth" - -DATASETS: - NAMES: ("MSMT17",) - TESTS: ("MSMT17",) - -DATALOADER: - NUM_WORKERS: 8 - -OUTPUT_DIR: "logs/dukemtmcreid/sbs101ibn-kd-sbs50ibn" \ No newline at end of file +INPUT: + SIZE_TRAIN: [ 256, 128 ] + SIZE_TEST: [ 256, 128 ] diff --git a/projects/DistillReID/configs/DukeMTMC/KD-bot101ibn-bot18ibn.yml b/projects/DistillReID/configs/DukeMTMC/KD-bot101ibn-bot18ibn.yml new file mode 100644 index 0000000..09c471d --- /dev/null +++ b/projects/DistillReID/configs/DukeMTMC/KD-bot101ibn-bot18ibn.yml @@ -0,0 +1,20 @@ +_BASE_: "../Base-bot-kd.yml" + +MODEL_TEACHER: + BACKBONE: + DEPTH: "101x" + FEAT_DIM: 2048 + +MODEL: + BACKBONE: + DEPTH: "18x" + FEAT_DIM: 512 + + STUDENT_WEIGHTS: "" + TEACHER_WEIGHTS: "projects/DistillReID/logs/dukemtmc/bagtricks_R101-ibn" + +DATASETS: + NAMES: ("DukeMTMC",) + TESTS: ("DukeMTMC",) + +OUTPUT_DIR: "projects/DistillReID/logs/dukemtmc/bot101ibn-kd-bot18ibn" \ No newline at end of file diff --git a/projects/DistillReID/configs/DukeMTMC/KD-bot101ibn-bot50ibn.yml b/projects/DistillReID/configs/DukeMTMC/KD-bot101ibn-bot50ibn.yml new file mode 100644 index 0000000..8ccbc4a --- /dev/null +++ b/projects/DistillReID/configs/DukeMTMC/KD-bot101ibn-bot50ibn.yml @@ -0,0 +1,20 @@ +_BASE_: "../Base-bot-kd.yml" + +MODEL_TEACHER: + BACKBONE: + DEPTH: "101x" + FEAT_DIM: 2048 + +MODEL: + BACKBONE: + DEPTH: "50x" + FEAT_DIM: 2048 + + STUDENT_WEIGHTS: "" + TEACHER_WEIGHTS: "projects/DistillReID/logs/dukemtmc/bagtricks_R101-ibn" + +DATASETS: + NAMES: ("DukeMTMC",) + TESTS: ("DukeMTMC",) + +OUTPUT_DIR: "projects/DistillReID/logs/dukemtmc/dukemtmc/bot101ibn-kd-bot50ibn" \ No newline at end of file diff --git a/projects/DistillReID/configs/DukeMTMC/KD-bot50ibn-bot18ibn.yml b/projects/DistillReID/configs/DukeMTMC/KD-bot50ibn-bot18ibn.yml new file mode 100644 index 0000000..940de53 --- /dev/null +++ b/projects/DistillReID/configs/DukeMTMC/KD-bot50ibn-bot18ibn.yml @@ -0,0 +1,20 @@ +_BASE_: "../Base-bot-kd.yml" + +MODEL_TEACHER: + BACKBONE: + DEPTH: "50x" + FEAT_DIM: 512 + +MODEL: + BACKBONE: + DEPTH: "18x" + FEAT_DIM: 512 + + STUDENT_WEIGHTS: "" + TEACHER_WEIGHTS: "projects/DistillReID/logs/dukemtmc/bagtricks_R50-ibn/model_final.pth" + +DATASETS: + NAMES: ("DukeMTMC",) + TESTS: ("DukeMTMC",) + +OUTPUT_DIR: "projects/DistillReID/logs/dukemtmc/bot50ibn-kd-bot18ibn" \ No newline at end of file diff --git a/projects/DistillReID/configs/DukeMTMC/KD-sbs101ibn-sbs18ibn.yml b/projects/DistillReID/configs/DukeMTMC/KD-sbs101ibn-sbs18ibn.yml new file mode 100644 index 0000000..849b861 --- /dev/null +++ b/projects/DistillReID/configs/DukeMTMC/KD-sbs101ibn-sbs18ibn.yml @@ -0,0 +1,20 @@ +_BASE_: "../Base-sbs-kd.yml" + +MODEL_TEACHER: + BACKBONE: + DEPTH: "101x" + FEAT_DIM: 2048 + +MODEL: + BACKBONE: + DEPTH: "34x" + FEAT_DIM: 512 + + STUDENT_WEIGHTS: "" + TEACHER_WEIGHTS: "projects/DistillReID/logs/dukemtmc/sbs_R101-ibn/model_final.pth" + +DATASETS: + NAMES: ("DukeMTMC",) + TESTS: ("DukeMTMC",) + +OUTPUT_DIR: "projects/DistillReID/logs/dukemtmc/sbs101ibn-kd-sbs18ibn" \ No newline at end of file diff --git a/projects/DistillReID/configs/DukeMTMC/KD-sbs101ibn-sbs50ibn.yml b/projects/DistillReID/configs/DukeMTMC/KD-sbs101ibn-sbs50ibn.yml new file mode 100644 index 0000000..01938aa --- /dev/null +++ b/projects/DistillReID/configs/DukeMTMC/KD-sbs101ibn-sbs50ibn.yml @@ -0,0 +1,20 @@ +_BASE_: "../Base-sbs-kd.yml" + +MODEL_TEACHER: + BACKBONE: + DEPTH: "101x" + FEAT_DIM: 2048 + +MODEL: + BACKBONE: + DEPTH: "50x" + FEAT_DIM: 2048 + + STUDENT_WEIGHTS: "" + TEACHER_WEIGHTS: "projects/DistillReID/logs/dukemtmc/sbs_R101-ibn/model_final.pth" + +DATASETS: + NAMES: ("DukeMTMC",) + TESTS: ("DukeMTMC",) + +OUTPUT_DIR: "projects/DistillReID/logs/dukemtmc/sbs101ibn-kd-sbs50ibn" \ No newline at end of file diff --git a/projects/DistillReID/configs/DukeMTMC/KD-sbs50ibn-sbs18ibn.yml b/projects/DistillReID/configs/DukeMTMC/KD-sbs50ibn-sbs18ibn.yml new file mode 100644 index 0000000..2c73527 --- /dev/null +++ b/projects/DistillReID/configs/DukeMTMC/KD-sbs50ibn-sbs18ibn.yml @@ -0,0 +1,20 @@ +_BASE_: "../Base-sbs-kd.yml" + +MODEL_TEACHER: + BACKBONE: + DEPTH: "50x" + FEAT_DIM: 2048 + +MODEL: + BACKBONE: + DEPTH: "18x" + FEAT_DIM: 512 + + STUDENT_WEIGHTS: "" + TEACHER_WEIGHTS: "projects/DistillReID/logs/dukemtmc/sbs_R50-ibn/model_final.pth" + +DATASETS: + NAMES: ("DukeMTMC",) + TESTS: ("DukeMTMC",) + +OUTPUT_DIR: "projects/DistillReID/logs/dukemtmc/sbs50ibn-kd-sbs18ibn" \ No newline at end of file diff --git a/projects/DistillReID/configs/DukeMTMC/bot101ibn.yml b/projects/DistillReID/configs/DukeMTMC/bot101ibn.yml new file mode 100644 index 0000000..4fccde2 --- /dev/null +++ b/projects/DistillReID/configs/DukeMTMC/bot101ibn.yml @@ -0,0 +1,12 @@ +_BASE_: "../../../../configs/Base-bagtricks.yml" + +MODEL: + BACKBONE: + DEPTH: "101x" + WITH_IBN: True + +DATASETS: + NAMES: ("DukeMTMC",) + TESTS: ("DukeMTMC",) + +OUTPUT_DIR: "projects/DistillReID/logs/dukemtmc/bagtricks_R101-ibn" \ No newline at end of file diff --git a/projects/DistillReID/configs/DukeMTMC/bot18ibn.yml b/projects/DistillReID/configs/DukeMTMC/bot18ibn.yml new file mode 100644 index 0000000..7976304 --- /dev/null +++ b/projects/DistillReID/configs/DukeMTMC/bot18ibn.yml @@ -0,0 +1,13 @@ +_BASE_: "../../../../configs/Base-bagtricks.yml" + +MODEL: + BACKBONE: + DEPTH: "18x" + WITH_IBN: True + FEAT_DIM: 512 + +DATASETS: + NAMES: ("DukeMTMC",) + TESTS: ("DukeMTMC",) + +OUTPUT_DIR: "projects/DistillReID/logs/dukemtmc/bagtricks_R18-ibn" \ No newline at end of file diff --git a/projects/DistillReID/configs/DukeMTMC/bot50ibn.yml b/projects/DistillReID/configs/DukeMTMC/bot50ibn.yml new file mode 100644 index 0000000..dbac555 --- /dev/null +++ b/projects/DistillReID/configs/DukeMTMC/bot50ibn.yml @@ -0,0 +1,12 @@ +_BASE_: "../../../../configs/Base-bagtricks.yml" + +MODEL: + BACKBONE: + DEPTH: "50x" + WITH_IBN: True + +DATASETS: + NAMES: ("DukeMTMC",) + TESTS: ("DukeMTMC",) + +OUTPUT_DIR: "projects/DistillReID/logs/dukemtmc/bagtricks_R50-ibn" \ No newline at end of file diff --git a/projects/DistillReID/configs-sbs-ibn-dukemtmcreid/sbs_R101-ibn.yml b/projects/DistillReID/configs/DukeMTMC/sbs101ibn.yml similarity index 66% rename from projects/DistillReID/configs-sbs-ibn-dukemtmcreid/sbs_R101-ibn.yml rename to projects/DistillReID/configs/DukeMTMC/sbs101ibn.yml index 21898f0..c06d62c 100644 --- a/projects/DistillReID/configs-sbs-ibn-dukemtmcreid/sbs_R101-ibn.yml +++ b/projects/DistillReID/configs/DukeMTMC/sbs101ibn.yml @@ -4,11 +4,10 @@ MODEL: BACKBONE: DEPTH: "101x" WITH_IBN: True - HEADS: - CLS_LAYER: "circleSoftmax" + FEAT_DIM: 2048 DATASETS: NAMES: ("DukeMTMC",) TESTS: ("DukeMTMC",) -OUTPUT_DIR: "logs/dukemtmcreid/sbs_R101-ibn" +OUTPUT_DIR: "projects/DistillReID/logs/dukemtmc/sbs_R101-ibn" \ No newline at end of file diff --git a/projects/DistillReID/configs-sbs-ibn-dukemtmcreid/sbs_R18-ibn.yml b/projects/DistillReID/configs/DukeMTMC/sbs18ibn.yml similarity index 58% rename from projects/DistillReID/configs-sbs-ibn-dukemtmcreid/sbs_R18-ibn.yml rename to projects/DistillReID/configs/DukeMTMC/sbs18ibn.yml index 3c2f920..27e8117 100644 --- a/projects/DistillReID/configs-sbs-ibn-dukemtmcreid/sbs_R18-ibn.yml +++ b/projects/DistillReID/configs/DukeMTMC/sbs18ibn.yml @@ -2,15 +2,12 @@ _BASE_: "../../../configs/Base-Strongerbaseline.yml" MODEL: BACKBONE: - WITH_NL: False DEPTH: "18x" WITH_IBN: True - HEADS: - IN_FEAT: 512 - CLS_LAYER: "circleSoftmax" + FEAT_DIM: 512 DATASETS: NAMES: ("DukeMTMC",) TESTS: ("DukeMTMC",) -OUTPUT_DIR: "logs/dukemtmcreid/sbs_R18-ibn" +OUTPUT_DIR: "projects/DistillReID/logs/dukemtmc/sbs_R18-ibn" \ No newline at end of file diff --git a/projects/DistillReID/configs-sbs-ibn-dukemtmcreid/sbs_R50-ibn.yml b/projects/DistillReID/configs/DukeMTMC/sbs50ibn.yml similarity index 57% rename from projects/DistillReID/configs-sbs-ibn-dukemtmcreid/sbs_R50-ibn.yml rename to projects/DistillReID/configs/DukeMTMC/sbs50ibn.yml index 21264fa..da4d8d5 100644 --- a/projects/DistillReID/configs-sbs-ibn-dukemtmcreid/sbs_R50-ibn.yml +++ b/projects/DistillReID/configs/DukeMTMC/sbs50ibn.yml @@ -2,15 +2,12 @@ _BASE_: "../../../configs/Base-Strongerbaseline.yml" MODEL: BACKBONE: - WITH_NL: False DEPTH: "50x" WITH_IBN: True - HEADS: - IN_FEAT: 2048 - CLS_LAYER: "circleSoftmax" + FEAT_DIM: 2048 DATASETS: NAMES: ("DukeMTMC",) TESTS: ("DukeMTMC",) -OUTPUT_DIR: "logs/dukemtmcreid/sbs_R50-ibn" +OUTPUT_DIR: "projects/DistillReID/logs/dukemtmc/sbs_R50-ibn" \ No newline at end of file diff --git a/projects/DistillReID/kdreid/config.py b/projects/DistillReID/kdreid/config.py index f376e6e..70abc3b 100644 --- a/projects/DistillReID/kdreid/config.py +++ b/projects/DistillReID/kdreid/config.py @@ -32,6 +32,8 @@ def add_kdreid_config(cfg): _C.MODEL_TEACHER.BACKBONE.WITH_SE = False # If use Non-local block in backbone _C.MODEL_TEACHER.BACKBONE.WITH_NL = False + # Input feature dimension + _C.MODEL_TEACHER.BACKBONE.FEAT_DIM = 2048 # for shufflenet _C.MODEL_TEACHER.BACKBONE.MODEL_SIZE = '1.0x' @@ -44,10 +46,8 @@ def add_kdreid_config(cfg): # teacher model HEADS options # ---------------------------------------------------------------------------- # _C.MODEL_TEACHER.HEADS = CN() - _C.MODEL_TEACHER.HEADS.NAME = "BNneckHead" + _C.MODEL_TEACHER.HEADS.NAME = "EmbeddingHead" - # Input feature dimension - _C.MODEL_TEACHER.HEADS.IN_FEAT = 2048 # Pooling layer type _C.MODEL_TEACHER.HEADS.POOL_LAYER = "avgpool" _C.MODEL_TEACHER.HEADS.NECK_FEAT = "before" @@ -82,23 +82,21 @@ def update_model_teacher_config(cfg): cfg.MODEL.BACKBONE.WITH_SE = cfg.MODEL_TEACHER.BACKBONE.WITH_SE # If use Non-local block in backbone cfg.MODEL.BACKBONE.WITH_NL = cfg.MODEL_TEACHER.BACKBONE.WITH_NL + # Input feature dimension + cfg.MODEL.BACKBONE.FEAT_DIM = cfg.MODEL_TEACHER.BACKBONE.FEAT_DIM cfg.MODEL.BACKBONE.PRETRAIN = False # for shufflenet cfg.MODEL.BACKBONE.MODEL_SIZE = cfg.MODEL_TEACHER.BACKBONE.MODEL_SIZE - # ---------------------------------------------------------------------------- # # teacher model HEADS options # ---------------------------------------------------------------------------- # cfg.MODEL.HEADS.NAME = cfg.MODEL_TEACHER.HEADS.NAME - # Input feature dimension - cfg.MODEL.HEADS.IN_FEAT = cfg.MODEL_TEACHER.HEADS.IN_FEAT # Pooling layer type cfg.MODEL.HEADS.POOL_LAYER = cfg.MODEL_TEACHER.HEADS.POOL_LAYER - # cfg.MODEL.HEADS.SCALE = cfg.MODEL_TEACHER.HEADS.SCALE cfg.MODEL.HEADS.MARGIN = cfg.MODEL_TEACHER.HEADS.MARGIN diff --git a/projects/DistillReID/kdreid/kd_trainer.py b/projects/DistillReID/kdreid/kd_trainer.py index d633144..e54925d 100644 --- a/projects/DistillReID/kdreid/kd_trainer.py +++ b/projects/DistillReID/kdreid/kd_trainer.py @@ -34,23 +34,13 @@ class KDTrainer(DefaultTrainer): for param in model_t.parameters(): param.requires_grad = False - logger = logging.getLogger('fastreid.'+__name__) + logger = logging.getLogger('fastreid.' + __name__) # Load pre-trained teacher model - print('88888\n'*10) logger.info("Loading teacher model ...") Checkpointer(model_t).load(cfg.MODEL.TEACHER_WEIGHTS) - # Load pre-trained student model - # logger.info("Loading student model ...") - # Checkpointer(self.model, self.data_loader.dataset).load(cfg.MODEL.STUDENT_WEIGHTS) - - def set_bn_eval(m): - classname = m.__class__.__name__ - if classname.find('BatchNorm') != -1: - m.eval() - - model_t.apply(set_bn_eval) + model_t.eval() self.model_t = model_t @@ -67,26 +57,29 @@ class KDTrainer(DefaultTrainer): data_time = time.perf_counter() - start - outputs, targets = self.model(data) + outs = self.model(data) # Compute reid loss if isinstance(self.model, DistributedDataParallel): - loss_dict = self.model.module.losses(outputs, targets) + loss_dict = self.model.module.losses(outs) else: - loss_dict = self.model.losses(outputs, targets) + loss_dict = self.model.losses(outs) with torch.no_grad(): - outputs_t, _ = self.model_t(data) + outs_t = self.model_t(data) - loss_dict['loss_kl'] = self.distill_loss(outputs[1], outputs_t[1].detach()) - # loss_dict['loss_pkt'] = 1e4 * self.pkt_loss(outputs[1], outputs_t[1].detach()) + q_logits = outs["outputs"]["pred_class_logits"] + t_logits = outs_t["outputs"]["pred_class_logits"].detach() + # k_logits = outs_k["outputs"]["pred_class_logits"].detach() + loss_dict['loss_kl'] = self.distill_loss(q_logits, t_logits, t=16) - losses = sum(loss for loss in loss_dict.values()) - self._detect_anomaly(losses, loss_dict) + losses = sum(loss_dict.values()) - metrics_dict = loss_dict - metrics_dict["data_time"] = data_time - self._write_metrics(metrics_dict) + with torch.cuda.stream(torch.cuda.Stream()): + metrics_dict = loss_dict + metrics_dict["data_time"] = data_time + self._write_metrics(metrics_dict) + self._detect_anomaly(losses, loss_dict) """ If you need accumulate gradients or something similar, you can @@ -139,4 +132,4 @@ class KDTrainer(DefaultTrainer): p_s = F.log_softmax(y_s / t, dim=1) p_t = F.softmax(y_t / t, dim=1) loss = F.kl_div(p_s, p_t, reduction='sum') * (t ** 2) / y_s.shape[0] - return loss \ No newline at end of file + return loss diff --git a/projects/DistillReID/kdreid/modeling/backbones/shufflenetv2/network.py b/projects/DistillReID/kdreid/modeling/backbones/shufflenetv2/network.py index 705a691..54fe0ea 100644 --- a/projects/DistillReID/kdreid/modeling/backbones/shufflenetv2/network.py +++ b/projects/DistillReID/kdreid/modeling/backbones/shufflenetv2/network.py @@ -7,11 +7,13 @@ import torch import torch.nn as nn from .blocks import ShuffleV2Block + class ShuffleNetV2(nn.Module): """ Reference: https://github.com/megvii-model/ShuffleNet-Series/tree/master/ShuffleNetV2 """ + def __init__(self, input_size=224, n_class=1000, model_size='1.5x'): super(ShuffleNetV2, self).__init__() print('model size is ', model_size) @@ -111,4 +113,4 @@ if __name__ == "__main__": test_data = torch.rand(5, 3, 224, 224) test_outputs = model(test_data) - print(test_outputs.size()) \ No newline at end of file + print(test_outputs.size()) diff --git a/projects/DistillReID/run_bagtricks.sh b/projects/DistillReID/run_bagtricks.sh deleted file mode 100644 index b8e3133..0000000 --- a/projects/DistillReID/run_bagtricks.sh +++ /dev/null @@ -1,132 +0,0 @@ -CUDA=3 - -# dukemtmcreid - -## bagtricks_R101-ibn.yml -#CUDA_VISIBLE_DEVICES=$CUDA python ./tools/train_net.py --config-file ./projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/bagtricks_R101-ibn.yml MODEL.DEVICE "cuda:0" -# -## bagtricks_R50-ibn.yml -#CUDA_VISIBLE_DEVICES=$CUDA python ./tools/train_net.py --config-file ./projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/bagtricks_R50-ibn.yml MODEL.DEVICE "cuda:0" -# -## bagtricks_R34-ibn.yml -#CUDA_VISIBLE_DEVICES=$CUDA python ./tools/train_net.py --config-file ./projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/bagtricks_R34-ibn.yml MODEL.DEVICE "cuda:0" -# -## bagtricks_R18-ibn.yml -#CUDA_VISIBLE_DEVICES=$CUDA python ./tools/train_net.py --config-file ./projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/bagtricks_R18-ibn.yml MODEL.DEVICE "cuda:0" - -# KD-bot101ibn-bot50ibn.yml -CUDA_VISIBLE_DEVICES=$CUDA python ./projects/DistillReID/train_net.py --kd --config-file ./projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot50ibn.yml MODEL.DEVICE "cuda:0" - -## KD-bot101ibn-bot34ibn.yml -#CUDA_VISIBLE_DEVICES=$CUDA python ./projects/DistillReID/train_net.py --kd --config-file ./projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot34ibn.yml MODEL.DEVICE "cuda:0" -# -## KD-bot101ibn-bot18ibn.yml -#CUDA_VISIBLE_DEVICES=$CUDA python ./projects/DistillReID/train_net.py --kd --config-file ./projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot18ibn.yml MODEL.DEVICE "cuda:0" -# -## KD-bot50ibn-bot34ibn.yml -#CUDA_VISIBLE_DEVICES=$CUDA python ./projects/DistillReID/train_net.py --kd --config-file ./projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot50ibn-bot34ibn.yml MODEL.DEVICE "cuda:0" -# -## KD-bot50ibn-bot18ibn.yml -#CUDA_VISIBLE_DEVICES=$CUDA python ./projects/DistillReID/train_net.py --kd --config-file ./projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot50ibn-bot18ibn.yml MODEL.DEVICE "cuda:0" -# -## KD-bot34ibn-bot18ibn.yml -#CUDA_VISIBLE_DEVICES=$CUDA python ./projects/DistillReID/train_net.py --kd --config-file ./projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot34ibn-bot18ibn.yml MODEL.DEVICE "cuda:0" -# -## KD-bot101ibn-bot50.yml -#CUDA_VISIBLE_DEVICES=$CUDA python ./projects/DistillReID/train_net.py --kd --config-file ./projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot50.yml MODEL.DEVICE "cuda:0" -# -## KD-bot101ibn-bot34.yml -#CUDA_VISIBLE_DEVICES=$CUDA python ./projects/DistillReID/train_net.py --kd --config-file ./projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot34.yml MODEL.DEVICE "cuda:0" -# -## KD-bot101ibn-bot18.yml -#CUDA_VISIBLE_DEVICES=$CUDA python ./projects/DistillReID/train_net.py --kd --config-file ./projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot18.yml MODEL.DEVICE "cuda:0" -# -# -# -# -## market1501 -# -### bagtricks_R101-ibn.yml -##CUDA_VISIBLE_DEVICES=$CUDA python ./tools/train_net.py --config-file ./projects/DistillReID/configs-bagtricks-ibn-market1501/bagtricks_R101-ibn.yml MODEL.DEVICE "cuda:0" -## -### bagtricks_R50-ibn.yml -##CUDA_VISIBLE_DEVICES=$CUDA python ./tools/train_net.py --config-file ./projects/DistillReID/configs-bagtricks-ibn-market1501/bagtricks_R50-ibn.yml MODEL.DEVICE "cuda:0" -## -### bagtricks_R34-ibn.yml -##CUDA_VISIBLE_DEVICES=$CUDA python ./tools/train_net.py --config-file ./projects/DistillReID/configs-bagtricks-ibn-market1501/bagtricks_R34-ibn.yml MODEL.DEVICE "cuda:0" -## -### bagtricks_R18-ibn.yml -##CUDA_VISIBLE_DEVICES=$CUDA python ./tools/train_net.py --config-file ./projects/DistillReID/configs-bagtricks-ibn-market1501/bagtricks_R18-ibn.yml MODEL.DEVICE "cuda:0" -# -## KD-bot101ibn-bot50ibn.yml -#CUDA_VISIBLE_DEVICES=$CUDA python ./projects/DistillReID/train_net.py --kd --config-file ./projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot50ibn.yml MODEL.DEVICE "cuda:0" -# -## KD-bot101ibn-bot34ibn.yml -#CUDA_VISIBLE_DEVICES=$CUDA python ./projects/DistillReID/train_net.py --kd --config-file ./projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot34ibn.yml MODEL.DEVICE "cuda:0" -# -## KD-bot101ibn-bot18ibn.yml -#CUDA_VISIBLE_DEVICES=$CUDA python ./projects/DistillReID/train_net.py --kd --config-file ./projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot18ibn.yml MODEL.DEVICE "cuda:0" -# -## KD-bot50ibn-bot34ibn.yml -#CUDA_VISIBLE_DEVICES=$CUDA python ./projects/DistillReID/train_net.py --kd --config-file ./projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot50ibn-bot34ibn.yml MODEL.DEVICE "cuda:0" -# -## KD-bot50ibn-bot18ibn.yml -#CUDA_VISIBLE_DEVICES=$CUDA python ./projects/DistillReID/train_net.py --kd --config-file ./projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot50ibn-bot18ibn.yml MODEL.DEVICE "cuda:0" -# -## KD-bot34ibn-bot18ibn.yml -#CUDA_VISIBLE_DEVICES=$CUDA python ./projects/DistillReID/train_net.py --kd --config-file ./projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot34ibn-bot18ibn.yml MODEL.DEVICE "cuda:0" -# -## KD-bot101ibn-bot50.yml -#CUDA_VISIBLE_DEVICES=$CUDA python ./projects/DistillReID/train_net.py --kd --config-file ./projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot50.yml MODEL.DEVICE "cuda:0" -# -## KD-bot101ibn-bot34.yml -#CUDA_VISIBLE_DEVICES=$CUDA python ./projects/DistillReID/train_net.py --kd --config-file ./projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot34.yml MODEL.DEVICE "cuda:0" -# -## KD-bot101ibn-bot18.yml -#CUDA_VISIBLE_DEVICES=$CUDA python ./projects/DistillReID/train_net.py --kd --config-file ./projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot18.yml MODEL.DEVICE "cuda:0" -# -# -# -# -# -## msmt17 -# -### bagtricks_R101-ibn.yml -##CUDA_VISIBLE_DEVICES=$CUDA python ./tools/train_net.py --config-file ./projects/DistillReID/configs-bagtricks-ibn-msmt17/bagtricks_R101-ibn.yml MODEL.DEVICE "cuda:0" -## -### bagtricks_R50-ibn.yml -##CUDA_VISIBLE_DEVICES=$CUDA python ./tools/train_net.py --config-file ./projects/DistillReID/configs-bagtricks-ibn-msmt17/bagtricks_R50-ibn.yml MODEL.DEVICE "cuda:0" -## -### bagtricks_R34-ibn.yml -##CUDA_VISIBLE_DEVICES=$CUDA python ./tools/train_net.py --config-file ./projects/DistillReID/configs-bagtricks-ibn-msmt17/bagtricks_R34-ibn.yml MODEL.DEVICE "cuda:0" -## -### bagtricks_R18-ibn.yml -##CUDA_VISIBLE_DEVICES=$CUDA python ./tools/train_net.py --config-file ./projects/DistillReID/configs-bagtricks-ibn-msmt17/bagtricks_R18-ibn.yml MODEL.DEVICE "cuda:0" -# -## KD-bot101ibn-bot50ibn.yml -#CUDA_VISIBLE_DEVICES=$CUDA python ./projects/DistillReID/train_net.py --kd --config-file ./projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot50ibn.yml MODEL.DEVICE "cuda:0" -# -## KD-bot101ibn-bot34ibn.yml -#CUDA_VISIBLE_DEVICES=$CUDA python ./projects/DistillReID/train_net.py --kd --config-file ./projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot34ibn.yml MODEL.DEVICE "cuda:0" -# -## KD-bot101ibn-bot18ibn.yml -#CUDA_VISIBLE_DEVICES=$CUDA python ./projects/DistillReID/train_net.py --kd --config-file ./projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot18ibn.yml MODEL.DEVICE "cuda:0" -# -## KD-bot50ibn-bot34ibn.yml -#CUDA_VISIBLE_DEVICES=$CUDA python ./projects/DistillReID/train_net.py --kd --config-file ./projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot50ibn-bot34ibn.yml MODEL.DEVICE "cuda:0" -# -## KD-bot50ibn-bot18ibn.yml -#CUDA_VISIBLE_DEVICES=$CUDA python ./projects/DistillReID/train_net.py --kd --config-file ./projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot50ibn-bot18ibn.yml MODEL.DEVICE "cuda:0" -# -## KD-bot34ibn-bot18ibn.yml -#CUDA_VISIBLE_DEVICES=$CUDA python ./projects/DistillReID/train_net.py --kd --config-file ./projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot34ibn-bot18ibn.yml MODEL.DEVICE "cuda:0" -# -## KD-bot101ibn-bot50.yml -#CUDA_VISIBLE_DEVICES=$CUDA python ./projects/DistillReID/train_net.py --kd --config-file ./projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot50.yml MODEL.DEVICE "cuda:0" -# -## KD-bot101ibn-bot34.yml -#CUDA_VISIBLE_DEVICES=$CUDA python ./projects/DistillReID/train_net.py --kd --config-file ./projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot34.yml MODEL.DEVICE "cuda:0" -# -## KD-bot101ibn-bot18.yml -#CUDA_VISIBLE_DEVICES=$CUDA python ./projects/DistillReID/train_net.py --kd --config-file ./projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot18.yml MODEL.DEVICE "cuda:0" -#