From 7ddb5b6f593adf3ae2e9ce2768c79891e7d95854 Mon Sep 17 00:00:00 2001 From: wangguanan Date: Mon, 10 Aug 2020 10:31:41 +0800 Subject: [PATCH] add msmt17 results --- projects/DistillReID/README.md | 8 +- run_ibna.sh | 164 +++++++++++++++++++++++++-------- 2 files changed, 132 insertions(+), 40 deletions(-) diff --git a/projects/DistillReID/README.md b/projects/DistillReID/README.md index 1b7e5e9..ad04360 100644 --- a/projects/DistillReID/README.md +++ b/projects/DistillReID/README.md @@ -34,12 +34,14 @@ All the experiments are conducted with a P40 GPU and ### Market-1501 -
Rank-1 (mAP) /
Q.Time/batch(128)		Student (BagTricks)
IBN-101IBN-50IBN-34IBN-18
Teacher
(BagTricks)		IBN-10195.43(88.95)/0.2698s95.19(89.52)/0.1791s94.51(87.82)/0.0869s93.85(85.77)/0.0612s
IBN-50-95.25(88.16)/0.1823s95.13(87.28)/0.0863s93.44(82.92)/0.0621s
IBN-34-94.63(84.91)/0.0860s
IBN-18---92.87(81.22)/0.0615s
Average Q.Time0.2698s0.1807s0.0864s0.0616s
+
Rank-1 (mAP) /
Q.Time/batch(128)		Student (BagTricks)
IBN-101IBN-50IBN-34IBN-18
Teacher
(BagTricks)		IBN-10195.43(88.95)/0.2698s95.19(89.52)/0.1791s94.51(87.82)/0.0869s93.85(85.77)/0.0612s
IBN-50-95.25(88.16)/0.1823s95.13(87.28)/0.0863s
IBN-34-94.63(84.91)/0.0860s93.71(85.20)/0.0620s
IBN-18---92.87(81.22)/0.0615s
Average Q.Time0.2698s0.1807s0.0864s0.0616s
### MSMT17 -Comming Soon +
Rank-1 (mAP) /
Q.Time/batch(128)		Student (BagTricks)
IBN-101IBN-50IBN-34IBN-18
Teacher
(BagTricks)		IBN-10181.95(60.51)/0.2693s82.37(62.08)/0.1792s81.07(58.56)/0.0872s77.77(52.77)/0.0610s
IBN-50-80.18(57.80)/0.1789s81.28(58.27)/0.0863s78.11(53.10)/0.0623s
IBN-34-78.27(53.41)/0.0873s77.65(52.82)/0.0615s
IBN-18---74.11(47.26)/0.0621s
Average Q.Time0.2693s0.1801s0.0868s0.0617s
## Contact -This project is conducted by [Guan'an Wang](https://wangguanan.github.io/) (guan.wang0706@gmail) and [Liao Xingyu](https://github.com/L1aoXingyu). \ No newline at end of file +This project is conducted by [Guan'an Wang](https://wangguanan.github.io/) (guan.wang0706@gmail) and [Liao Xingyu](https://github.com/L1aoXingyu). + + diff --git a/run_ibna.sh b/run_ibna.sh index 793c0d3..b8e3133 100644 --- a/run_ibna.sh +++ b/run_ibna.sh @@ -1,42 +1,132 @@ -CUDA=0 +CUDA=3 -# IBN-Net +# dukemtmcreid -# bagtricks_R101-ibn.yml -#CUDA_VISIBLE_DEVICES=$CUDA python ./tools/train_net.py --config-file ./projects/DistillReID/configs-ibn/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-ibn/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-ibn/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-ibn/bagtricks_R18-ibn.yml MODEL.DEVICE "cuda:0" +## 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-ibn/KD-bot101ibn-bot50ibn.yml MODEL.DEVICE "cuda:0" +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-ibn/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-ibn/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-ibn/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-ibn/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-ibn/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-ibn/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-ibn/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-ibn/KD-bot101ibn-bot18.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" +#