mirrors
/
fast-reid
mirror of https://github.com/JDAI-CV/fast-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

update distill reid project (#287) 

Summary: refactor distill reid code and config files
pull/299/head
liaoxingyu 2020-09-30 16:05:58 +08:00
parent 3d1bae9f13
commit 69f4407267
84 changed files with 233 additions and 2791 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
projects/DistillReID/README.md
View File

@ -15,9 +15,9 @@ This project provides a training script of small model
# a demo on DukeMTMC-reID dataset # a demo on DukeMTMC-reID dataset
# please see more in ./configs # please see more in ./configs
# train BagTricksIBN50 as teacher model # 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 # 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 ## Experimental Results and Trained Models
@ -42,6 +42,6 @@ All the experiments are conducted with a P40 GPU and
## Contact ## 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).

53
projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot18.yml
View File

@ -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"

53
projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot18ibn.yml
View File

@ -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"

53
projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot34.yml
View File

@ -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"

53
projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot34ibn.yml
View File

@ -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"

53
projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot50.yml
View File

@ -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"

53
projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot101ibn-bot50ibn.yml
View File

@ -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"

53
projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot34ibn-bot18ibn.yml
View File

@ -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"

53
projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot50ibn-bot18ibn.yml
View File

@ -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"

53
projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/KD-bot50ibn-bot34ibn.yml
View File

@ -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"

12
projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/bagtricks_R101-ibn.yml
View File

@ -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"

14
projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/bagtricks_R18-ibn.yml
View File

@ -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"

14
projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/bagtricks_R34-ibn.yml
View File

@ -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"

12
projects/DistillReID/configs-bagtricks-ibn-dukemtmcreid/bagtricks_R50-ibn.yml
View File

@ -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"

53
projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot18.yml
View File

@ -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"

53
projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot18ibn.yml
View File

@ -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"

53
projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot34.yml
View File

@ -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"

53
projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot34ibn.yml
View File

@ -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"

53
projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot50.yml
View File

@ -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"

53
projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot101ibn-bot50ibn.yml
View File

@ -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"

53
projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot34ibn-bot18ibn.yml
View File

@ -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"

53
projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot50ibn-bot18ibn.yml
View File

@ -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"

53
projects/DistillReID/configs-bagtricks-ibn-market1501/KD-bot50ibn-bot34ibn.yml
View File

@ -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"

12
projects/DistillReID/configs-bagtricks-ibn-market1501/bagtricks_R101-ibn.yml
View File

@ -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"

14
projects/DistillReID/configs-bagtricks-ibn-market1501/bagtricks_R18-ibn.yml
View File

@ -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"

14
projects/DistillReID/configs-bagtricks-ibn-market1501/bagtricks_R34-ibn.yml
View File

@ -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"

12
projects/DistillReID/configs-bagtricks-ibn-market1501/bagtricks_R50-ibn.yml
View File

@ -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"

53
projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot18.yml
View File

@ -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"

53
projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot18ibn.yml
View File

@ -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"

53
projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot34.yml
View File

@ -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"

53
projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot34ibn.yml
View File

@ -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"

53
projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot50.yml
View File

@ -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"

53
projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot101ibn-bot50ibn.yml
View File

@ -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"

53
projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot34ibn-bot18ibn.yml
View File

@ -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"

53
projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot50ibn-bot18ibn.yml
View File

@ -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"

53
projects/DistillReID/configs-bagtricks-ibn-msmt17/KD-bot50ibn-bot34ibn.yml
View File

@ -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"

12
projects/DistillReID/configs-bagtricks-ibn-msmt17/bagtricks_R101-ibn.yml
View File

@ -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"

14
projects/DistillReID/configs-bagtricks-ibn-msmt17/bagtricks_R18-ibn.yml
View File

@ -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"

14
projects/DistillReID/configs-bagtricks-ibn-msmt17/bagtricks_R34-ibn.yml
View File

@ -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"

12
projects/DistillReID/configs-bagtricks-ibn-msmt17/bagtricks_R50-ibn.yml
View File

@ -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"

50
projects/DistillReID/configs-sbs-ibn-dukemtmcreid/KD-sbs101ibn-sbs18ibn.yml
View File

@ -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"

51
projects/DistillReID/configs-sbs-ibn-dukemtmcreid/KD-sbs101ibn-sbs34ibn.yml
View File

@ -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"

51
projects/DistillReID/configs-sbs-ibn-dukemtmcreid/KD-sbs101ibn-sbs50ibn.yml
View File

@ -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"

50
projects/DistillReID/configs-sbs-ibn-dukemtmcreid/KD-sbs34ibn-sbs18ibn.yml
View File

@ -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"

50
projects/DistillReID/configs-sbs-ibn-dukemtmcreid/KD-sbs50ibn-sbs18ibn.yml
View File

@ -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"

51
projects/DistillReID/configs-sbs-ibn-dukemtmcreid/KD-sbs50ibn-sbs34ibn.yml
View File

@ -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"

16
projects/DistillReID/configs-sbs-ibn-dukemtmcreid/sbs_R34-ibn.yml
View File

@ -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"

50
projects/DistillReID/configs-sbs-ibn-market1501/KD-sbs101ibn-sbs18ibn.yml
View File

@ -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"

51
projects/DistillReID/configs-sbs-ibn-market1501/KD-sbs101ibn-sbs34ibn.yml
View File

@ -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"

51
projects/DistillReID/configs-sbs-ibn-market1501/KD-sbs101ibn-sbs50ibn.yml
View File

@ -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"

50
projects/DistillReID/configs-sbs-ibn-market1501/KD-sbs34ibn-sbs18ibn.yml
View File

@ -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"

50
projects/DistillReID/configs-sbs-ibn-market1501/KD-sbs50ibn-sbs18ibn.yml
View File

@ -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"

51
projects/DistillReID/configs-sbs-ibn-market1501/KD-sbs50ibn-sbs34ibn.yml
View File

@ -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"

14
projects/DistillReID/configs-sbs-ibn-market1501/sbs_R101-ibn.yml
View File

@ -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"

16
projects/DistillReID/configs-sbs-ibn-market1501/sbs_R18-ibn.yml
View File

@ -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"

16
projects/DistillReID/configs-sbs-ibn-market1501/sbs_R34-ibn.yml
View File

@ -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"

16
projects/DistillReID/configs-sbs-ibn-market1501/sbs_R50-ibn.yml
View File

@ -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"

50
projects/DistillReID/configs-sbs-ibn-msmt17/KD-sbs101ibn-sbs18ibn.yml
View File

@ -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"

51
projects/DistillReID/configs-sbs-ibn-msmt17/KD-sbs101ibn-sbs34ibn.yml
View File

@ -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"

50
projects/DistillReID/configs-sbs-ibn-msmt17/KD-sbs34ibn-sbs18ibn.yml
View File

@ -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"

50
projects/DistillReID/configs-sbs-ibn-msmt17/KD-sbs50ibn-sbs18ibn.yml
View File

@ -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"

51
projects/DistillReID/configs-sbs-ibn-msmt17/KD-sbs50ibn-sbs34ibn.yml
View File

@ -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"

14
projects/DistillReID/configs-sbs-ibn-msmt17/sbs_R101-ibn.yml
View File

@ -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"

16
projects/DistillReID/configs-sbs-ibn-msmt17/sbs_R18-ibn.yml
View File

@ -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"

16
projects/DistillReID/configs-sbs-ibn-msmt17/sbs_R34-ibn.yml
View File

@ -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"

16
projects/DistillReID/configs-sbs-ibn-msmt17/sbs_R50-ibn.yml
View File

@ -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"

30
projects/DistillReID/configs/Base-bot-kd.yml 100644
View File

@ -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"

38
projects/DistillReID/configs-sbs-ibn-msmt17/KD-sbs101ibn-sbs50ibn.yml → projects/DistillReID/configs/Base-sbs-kd.yml
View File

@ -1,8 +1,5 @@
_BASE_: "../../../configs/Base-Strongerbaseline.yml" _BASE_: "../../../configs/Base-Strongerbaseline.yml"
SOLVER:
IMS_PER_BATCH: 64
MODEL_TEACHER: MODEL_TEACHER:
META_ARCHITECTURE: "Baseline" META_ARCHITECTURE: "Baseline"
@ -10,42 +7,31 @@ MODEL_TEACHER:
NAME: "build_resnet_backbone" NAME: "build_resnet_backbone"
NORM: "BN" NORM: "BN"
DEPTH: "101x" DEPTH: "101x"
FEAT_DIM: 2048
LAST_STRIDE: 1 LAST_STRIDE: 1
WITH_NL: True WITH_NL: False
WITH_IBN: True WITH_IBN: True
PRETRAIN: False PRETRAIN: True
HEADS: HEADS:
NAME: "BNneckHead" NAME: "EmbeddingHead"
IN_FEAT: 2048
NORM: "BN" NORM: "BN"
NECK_FEAT: "after" NECK_FEAT: "after"
POOL_LAYER: "gempool" POOL_LAYER: "gempoolP"
CLS_LAYER: "circle" CLS_LAYER: "circleSoftmax"
SCALE: 64 SCALE: 64
MARGIN: 0.35 MARGIN: 0.35
MODEL: MODEL:
META_ARCHITECTURE: "Baseline"
BACKBONE: BACKBONE:
WITH_NL: False NAME: "build_resnet_backbone"
DEPTH: "50x" DEPTH: "50x"
FEAT_DIM: 2048
WITH_IBN: True WITH_IBN: True
HEADS:
IN_FEAT: 2048
CLS_LAYER: "circleSoftmax"
STUDENT_WEIGHTS: "" STUDENT_WEIGHTS: ""
TEACHER_WEIGHTS: "./logs/dukemtmcreid/sbs_R101-ibn/model_final.pth" TEACHER_WEIGHTS: "logs/dukemtmc/bagtricks_R34-ibn/model_final.pth"
INPUT:
DATASETS: SIZE_TRAIN: [ 256, 128 ]
NAMES: ("MSMT17",) SIZE_TEST: [ 256, 128 ]
TESTS: ("MSMT17",)
DATALOADER:
NUM_WORKERS: 8
OUTPUT_DIR: "logs/dukemtmcreid/sbs101ibn-kd-sbs50ibn"

20
projects/DistillReID/configs/DukeMTMC/KD-bot101ibn-bot18ibn.yml 100644
View File

@ -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"

20
projects/DistillReID/configs/DukeMTMC/KD-bot101ibn-bot50ibn.yml 100644
View File

@ -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"

20
projects/DistillReID/configs/DukeMTMC/KD-bot50ibn-bot18ibn.yml 100644
View File

@ -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"

20
projects/DistillReID/configs/DukeMTMC/KD-sbs101ibn-sbs18ibn.yml 100644
View File

@ -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"

20
projects/DistillReID/configs/DukeMTMC/KD-sbs101ibn-sbs50ibn.yml 100644
View File

@ -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"

20
projects/DistillReID/configs/DukeMTMC/KD-sbs50ibn-sbs18ibn.yml 100644
View File

@ -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"

12
projects/DistillReID/configs/DukeMTMC/bot101ibn.yml 100644
View File

@ -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"

13
projects/DistillReID/configs/DukeMTMC/bot18ibn.yml 100644
View File

@ -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"

12
projects/DistillReID/configs/DukeMTMC/bot50ibn.yml 100644
View File

@ -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"

5
projects/DistillReID/configs-sbs-ibn-dukemtmcreid/sbs_R101-ibn.yml → projects/DistillReID/configs/DukeMTMC/sbs101ibn.yml
View File

@ -4,11 +4,10 @@ MODEL:
BACKBONE: BACKBONE:
DEPTH: "101x" DEPTH: "101x"
WITH_IBN: True WITH_IBN: True
HEADS: FEAT_DIM: 2048
CLS_LAYER: "circleSoftmax"
DATASETS: DATASETS:
NAMES: ("DukeMTMC",) NAMES: ("DukeMTMC",)
TESTS: ("DukeMTMC",) TESTS: ("DukeMTMC",)
OUTPUT_DIR: "logs/dukemtmcreid/sbs_R101-ibn" OUTPUT_DIR: "projects/DistillReID/logs/dukemtmc/sbs_R101-ibn"

7
projects/DistillReID/configs-sbs-ibn-dukemtmcreid/sbs_R18-ibn.yml → projects/DistillReID/configs/DukeMTMC/sbs18ibn.yml
View File

@ -2,15 +2,12 @@ _BASE_: "../../../configs/Base-Strongerbaseline.yml"
MODEL: MODEL:
BACKBONE: BACKBONE:
WITH_NL: False
DEPTH: "18x" DEPTH: "18x"
WITH_IBN: True WITH_IBN: True
HEADS: FEAT_DIM: 512
IN_FEAT: 512
CLS_LAYER: "circleSoftmax"
DATASETS: DATASETS:
NAMES: ("DukeMTMC",) NAMES: ("DukeMTMC",)
TESTS: ("DukeMTMC",) TESTS: ("DukeMTMC",)
OUTPUT_DIR: "logs/dukemtmcreid/sbs_R18-ibn" OUTPUT_DIR: "projects/DistillReID/logs/dukemtmc/sbs_R18-ibn"

7
projects/DistillReID/configs-sbs-ibn-dukemtmcreid/sbs_R50-ibn.yml → projects/DistillReID/configs/DukeMTMC/sbs50ibn.yml
View File

@ -2,15 +2,12 @@ _BASE_: "../../../configs/Base-Strongerbaseline.yml"
MODEL: MODEL:
BACKBONE: BACKBONE:
WITH_NL: False
DEPTH: "50x" DEPTH: "50x"
WITH_IBN: True WITH_IBN: True
HEADS: FEAT_DIM: 2048
IN_FEAT: 2048
CLS_LAYER: "circleSoftmax"
DATASETS: DATASETS:
NAMES: ("DukeMTMC",) NAMES: ("DukeMTMC",)
TESTS: ("DukeMTMC",) TESTS: ("DukeMTMC",)
OUTPUT_DIR: "logs/dukemtmcreid/sbs_R50-ibn" OUTPUT_DIR: "projects/DistillReID/logs/dukemtmc/sbs_R50-ibn"

12
projects/DistillReID/kdreid/config.py
View File

@ -32,6 +32,8 @@ def add_kdreid_config(cfg):
_C.MODEL_TEACHER.BACKBONE.WITH_SE = False _C.MODEL_TEACHER.BACKBONE.WITH_SE = False
# If use Non-local block in backbone # If use Non-local block in backbone
_C.MODEL_TEACHER.BACKBONE.WITH_NL = False _C.MODEL_TEACHER.BACKBONE.WITH_NL = False
# Input feature dimension
_C.MODEL_TEACHER.BACKBONE.FEAT_DIM = 2048
# for shufflenet # for shufflenet
_C.MODEL_TEACHER.BACKBONE.MODEL_SIZE = '1.0x' _C.MODEL_TEACHER.BACKBONE.MODEL_SIZE = '1.0x'
@ -44,10 +46,8 @@ def add_kdreid_config(cfg):
# teacher model HEADS options # teacher model HEADS options
# ---------------------------------------------------------------------------- # # ---------------------------------------------------------------------------- #
_C.MODEL_TEACHER.HEADS = CN() _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 # Pooling layer type
_C.MODEL_TEACHER.HEADS.POOL_LAYER = "avgpool" _C.MODEL_TEACHER.HEADS.POOL_LAYER = "avgpool"
_C.MODEL_TEACHER.HEADS.NECK_FEAT = "before" _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 cfg.MODEL.BACKBONE.WITH_SE = cfg.MODEL_TEACHER.BACKBONE.WITH_SE
# If use Non-local block in backbone # If use Non-local block in backbone
cfg.MODEL.BACKBONE.WITH_NL = cfg.MODEL_TEACHER.BACKBONE.WITH_NL 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 cfg.MODEL.BACKBONE.PRETRAIN = False
# for shufflenet # for shufflenet
cfg.MODEL.BACKBONE.MODEL_SIZE = cfg.MODEL_TEACHER.BACKBONE.MODEL_SIZE cfg.MODEL.BACKBONE.MODEL_SIZE = cfg.MODEL_TEACHER.BACKBONE.MODEL_SIZE
# ---------------------------------------------------------------------------- # # ---------------------------------------------------------------------------- #
# teacher model HEADS options # teacher model HEADS options
# ---------------------------------------------------------------------------- # # ---------------------------------------------------------------------------- #
cfg.MODEL.HEADS.NAME = cfg.MODEL_TEACHER.HEADS.NAME 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 # Pooling layer type
cfg.MODEL.HEADS.POOL_LAYER = cfg.MODEL_TEACHER.HEADS.POOL_LAYER cfg.MODEL.HEADS.POOL_LAYER = cfg.MODEL_TEACHER.HEADS.POOL_LAYER
#
cfg.MODEL.HEADS.SCALE = cfg.MODEL_TEACHER.HEADS.SCALE cfg.MODEL.HEADS.SCALE = cfg.MODEL_TEACHER.HEADS.SCALE
cfg.MODEL.HEADS.MARGIN = cfg.MODEL_TEACHER.HEADS.MARGIN cfg.MODEL.HEADS.MARGIN = cfg.MODEL_TEACHER.HEADS.MARGIN

39
projects/DistillReID/kdreid/kd_trainer.py
View File

@ -34,23 +34,13 @@ class KDTrainer(DefaultTrainer):
for param in model_t.parameters(): for param in model_t.parameters():
param.requires_grad = False param.requires_grad = False
logger = logging.getLogger('fastreid.'+__name__) logger = logging.getLogger('fastreid.' + __name__)
# Load pre-trained teacher model # Load pre-trained teacher model
print('88888\n'*10)
logger.info("Loading teacher model ...") logger.info("Loading teacher model ...")
Checkpointer(model_t).load(cfg.MODEL.TEACHER_WEIGHTS) Checkpointer(model_t).load(cfg.MODEL.TEACHER_WEIGHTS)
# Load pre-trained student model model_t.eval()
# 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)
self.model_t = model_t self.model_t = model_t
@ -67,26 +57,29 @@ class KDTrainer(DefaultTrainer):
data_time = time.perf_counter() - start data_time = time.perf_counter() - start
outputs, targets = self.model(data) outs = self.model(data)
# Compute reid loss # Compute reid loss
if isinstance(self.model, DistributedDataParallel): if isinstance(self.model, DistributedDataParallel):
loss_dict = self.model.module.losses(outputs, targets) loss_dict = self.model.module.losses(outs)
else: else:
loss_dict = self.model.losses(outputs, targets) loss_dict = self.model.losses(outs)
with torch.no_grad(): 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()) q_logits = outs["outputs"]["pred_class_logits"]
# loss_dict['loss_pkt'] = 1e4 * self.pkt_loss(outputs[1], outputs_t[1].detach()) 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()) losses = sum(loss_dict.values())
self._detect_anomaly(losses, loss_dict)
metrics_dict = loss_dict with torch.cuda.stream(torch.cuda.Stream()):
metrics_dict["data_time"] = data_time metrics_dict = loss_dict
self._write_metrics(metrics_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 If you need accumulate gradients or something similar, you can

2
projects/DistillReID/kdreid/modeling/backbones/shufflenetv2/network.py
View File

@ -7,11 +7,13 @@ import torch
import torch.nn as nn import torch.nn as nn
from .blocks import ShuffleV2Block from .blocks import ShuffleV2Block
class ShuffleNetV2(nn.Module): class ShuffleNetV2(nn.Module):
""" """
Reference: Reference:
https://github.com/megvii-model/ShuffleNet-Series/tree/master/ShuffleNetV2 https://github.com/megvii-model/ShuffleNet-Series/tree/master/ShuffleNetV2
""" """
def __init__(self, input_size=224, n_class=1000, model_size='1.5x'): def __init__(self, input_size=224, n_class=1000, model_size='1.5x'):
super(ShuffleNetV2, self).__init__() super(ShuffleNetV2, self).__init__()
print('model size is ', model_size) print('model size is ', model_size)

132
projects/DistillReID/run_bagtricks.sh
View File

@ -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"
#