From 64861ea6c85122bbf8577c10bd7fc6e551495b3a Mon Sep 17 00:00:00 2001 From: liaoxingyu Date: Wed, 29 Jul 2020 19:48:49 +0800 Subject: [PATCH] update config file --- configs/DukeMTMC/AGW_R101-ibn.yml | 2 +- configs/DukeMTMC/bagtricks_R101-ibn.yml | 2 +- configs/DukeMTMC/sbs_R101-ibn.yml | 2 +- configs/MSMT17/AGW_R101-ibn.yml | 2 +- configs/MSMT17/bagtricks_R101-ibn.yml | 2 +- configs/MSMT17/sbs_R101-ibn.yml | 2 +- configs/Market1501/AGW_R101-ibn.yml | 2 +- configs/Market1501/bagtricks_R101-ibn.yml | 2 +- configs/Market1501/sbs_R101-ibn.yml | 2 +- demo/run_bjz.sh | 9 +++++++++ demo/run_demo.sh | 6 +++--- tools/deploy/run_export.sh | 7 +++---- 12 files changed, 24 insertions(+), 16 deletions(-) create mode 100644 demo/run_bjz.sh diff --git a/configs/DukeMTMC/AGW_R101-ibn.yml b/configs/DukeMTMC/AGW_R101-ibn.yml index dcf852048..592ba0fe8 100644 --- a/configs/DukeMTMC/AGW_R101-ibn.yml +++ b/configs/DukeMTMC/AGW_R101-ibn.yml @@ -2,7 +2,7 @@ _BASE_: "../Base-AGW.yml" MODEL: BACKBONE: - DEPTH: 101 + DEPTH: "101x" WITH_IBN: True DATASETS: diff --git a/configs/DukeMTMC/bagtricks_R101-ibn.yml b/configs/DukeMTMC/bagtricks_R101-ibn.yml index ca3b4bc98..0865131aa 100644 --- a/configs/DukeMTMC/bagtricks_R101-ibn.yml +++ b/configs/DukeMTMC/bagtricks_R101-ibn.yml @@ -2,7 +2,7 @@ _BASE_: "../Base-bagtricks.yml" MODEL: BACKBONE: - DEPTH: 101 + DEPTH: "101x" WITH_IBN: True DATASETS: diff --git a/configs/DukeMTMC/sbs_R101-ibn.yml b/configs/DukeMTMC/sbs_R101-ibn.yml index b22aac00e..6e2c93fa0 100644 --- a/configs/DukeMTMC/sbs_R101-ibn.yml +++ b/configs/DukeMTMC/sbs_R101-ibn.yml @@ -2,7 +2,7 @@ _BASE_: "../Base-Strongerbaseline.yml" MODEL: BACKBONE: - DEPTH: 101 + DEPTH: "101x" WITH_IBN: True DATASETS: diff --git a/configs/MSMT17/AGW_R101-ibn.yml b/configs/MSMT17/AGW_R101-ibn.yml index 12f49a268..a122f6ca2 100644 --- a/configs/MSMT17/AGW_R101-ibn.yml +++ b/configs/MSMT17/AGW_R101-ibn.yml @@ -2,7 +2,7 @@ _BASE_: "../Base-AGW.yml" MODEL: BACKBONE: - DEPTH: 101 + DEPTH: "101x" WITH_IBN: True DATASETS: diff --git a/configs/MSMT17/bagtricks_R101-ibn.yml b/configs/MSMT17/bagtricks_R101-ibn.yml index aa21848c9..693269f05 100644 --- a/configs/MSMT17/bagtricks_R101-ibn.yml +++ b/configs/MSMT17/bagtricks_R101-ibn.yml @@ -2,7 +2,7 @@ _BASE_: "../Base-bagtricks.yml" MODEL: BACKBONE: - DEPTH: 101 + DEPTH: "101x" WITH_IBN: True DATASETS: diff --git a/configs/MSMT17/sbs_R101-ibn.yml b/configs/MSMT17/sbs_R101-ibn.yml index 85c7c1a28..f3b6b7e88 100644 --- a/configs/MSMT17/sbs_R101-ibn.yml +++ b/configs/MSMT17/sbs_R101-ibn.yml @@ -2,7 +2,7 @@ _BASE_: "../Base-Strongerbaseline.yml" MODEL: BACKBONE: - DEPTH: 101 + DEPTH: "101x" WITH_IBN: True DATASETS: diff --git a/configs/Market1501/AGW_R101-ibn.yml b/configs/Market1501/AGW_R101-ibn.yml index d86dbc884..94d3d85b2 100644 --- a/configs/Market1501/AGW_R101-ibn.yml +++ b/configs/Market1501/AGW_R101-ibn.yml @@ -2,7 +2,7 @@ _BASE_: "../Base-AGW.yml" MODEL: BACKBONE: - DEPTH: 101 + DEPTH: "101x" WITH_IBN: True DATASETS: diff --git a/configs/Market1501/bagtricks_R101-ibn.yml b/configs/Market1501/bagtricks_R101-ibn.yml index 03b513255..de11e3f0d 100644 --- a/configs/Market1501/bagtricks_R101-ibn.yml +++ b/configs/Market1501/bagtricks_R101-ibn.yml @@ -2,7 +2,7 @@ _BASE_: "../Base-bagtricks.yml" MODEL: BACKBONE: - DEPTH: "101" + DEPTH: "101x" WITH_IBN: True DATASETS: diff --git a/configs/Market1501/sbs_R101-ibn.yml b/configs/Market1501/sbs_R101-ibn.yml index ff740537b..13b9df039 100644 --- a/configs/Market1501/sbs_R101-ibn.yml +++ b/configs/Market1501/sbs_R101-ibn.yml @@ -2,7 +2,7 @@ _BASE_: "../Base-Strongerbaseline.yml" MODEL: BACKBONE: - DEPTH: 101 + DEPTH: "101x" WITH_IBN: True DATASETS: diff --git a/demo/run_bjz.sh b/demo/run_bjz.sh new file mode 100644 index 000000000..4b2163d43 --- /dev/null +++ b/demo/run_bjz.sh @@ -0,0 +1,9 @@ +python demo.py --config-file ../projects/bjzProject/logs/bjz/R50_512dim_circle_bjz_0618_8x3/config.yaml \ +--input \ +"/home/liaoxingyu2/lxy/logs/badcase/seed.jpg" \ +"/home/liaoxingyu2/lxy/logs/badcase/leaf_0.4883.jpg" \ +"/home/liaoxingyu2/lxy/logs/badcase/leaf_0.467.jpg" \ +--output logs/R50_256x128_pytorch_feat_output \ +--opts MODEL.WEIGHTS ../projects/bjzProject/logs/bjz/R50_512dim_circle_bjz_0618_8x32/model_final.pth \ +MODEL.DEVICE "cuda: 2" + diff --git a/demo/run_demo.sh b/demo/run_demo.sh index 1601aa013..6f3f28078 100644 --- a/demo/run_demo.sh +++ b/demo/run_demo.sh @@ -1,3 +1,3 @@ -python demo/visualize_result.py --config-file 'logs/dukemtmc/mgn_R50-ibn/config.yaml' \ ---parallel --vis-label --dataset-name 'DukeMTMC' --output 'logs/mgn_duke_vis' \ ---opts MODEL.WEIGHTS "logs/dukemtmc/mgn_R50-ibn/model_final.pth" +python demo/visualize_result.py --config-file logs/dukemtmc/mgn_R50-ibn/config.yaml \ +--parallel --vis-label --dataset-name 'DukeMTMC' --output logs/mgn_duke_vis \ +--opts MODEL.WEIGHTS logs/dukemtmc/mgn_R50-ibn/model_final.pth diff --git a/tools/deploy/run_export.sh b/tools/deploy/run_export.sh index fd3dfb171..48f5c9bf3 100644 --- a/tools/deploy/run_export.sh +++ b/tools/deploy/run_export.sh @@ -1,5 +1,4 @@ - -python caffe_export.py --config-file "/export/home/lxy/cvpalgo-fast-reid/logs/dukemtmc/R34/config.yaml" \ +python caffe_export.py --config-file /export/home/lxy/cvpalgo-fast-reid/logs/dukemtmc/R34/config.yaml \ --name "baseline_R34" \ ---output "logs/caffe_R34" \ ---opts MODEL.WEIGHTS "/export/home/lxy/cvpalgo-fast-reid/logs/dukemtmc/R34/model_final.pth" +--output logs/caffe_R34 \ +--opts MODEL.WEIGHTS /export/home/lxy/cvpalgo-fast-reid/logs/dukemtmc/R34/model_final.pth