From 28f50410a5c278e18f2df13fe6d14441f50bd386 Mon Sep 17 00:00:00 2001 From: darkliang <12132342@mail.sustech.edu.cn> Date: Tue, 12 Apr 2022 13:44:07 +0800 Subject: [PATCH] Reconstruct the directory structure --- config/{ => baseline}/baseline.yaml | 2 +- config/{ => baseline}/baseline_GREW.yaml | 2 +- config/{ => baseline}/baseline_OUMVLP.yaml | 2 +- .../HID => config/baseline}/baseline_hid.yaml | 2 +- config/default.yaml | 2 +- config/{ => gaitgl}/gaitgl.yaml | 2 +- config/{ => gaitgl}/gaitgl_OUMVLP.yaml | 2 +- config/{ => gaitpart}/gaitpart.yaml | 0 config/{ => gaitpart}/gaitpart_OUMVLP.yaml | 2 +- config/{ => gaitset}/gaitset.yaml | 0 config/{ => gaitset}/gaitset_OUMVLP.yaml | 2 +- config/gln/gln_phase1.yaml | 2 +- config/gln/gln_phase2.yaml | 2 +- .../CASIA-B}/CASIA-B.json | 0 .../CASIA-B}/CASIA-B_include_005.json | 0 datasets/CASIA-B/README.md | 27 ++++++++ {misc/partitions => datasets/GREW}/GREW.json | 0 {misc => datasets}/GREW/README.md | 13 ++-- {misc => datasets/GREW}/rearrange_GREW.py | 0 {misc => datasets}/HID/HID.json | 0 {misc => datasets}/HID/README.md | 0 {misc => datasets}/HID/pretreatment_HID.py | 0 .../OUMVLP}/OUMVLP.json | 0 datasets/OUMVLP/README.md | 67 +++++++++++++++++++ {misc => datasets/OUMVLP}/extractor.py | 0 {misc => datasets/OUMVLP}/rearrange_OUMVLP.py | 0 datasets/README.md | 24 +++++++ {misc => datasets}/pretreatment.py | 0 docs/0.prepare_dataset.md | 14 ++-- docs/3.advanced_usages.md | 8 +-- test.sh | 16 ++--- train.sh | 16 ++--- 32 files changed, 162 insertions(+), 45 deletions(-) rename config/{ => baseline}/baseline.yaml (97%) rename config/{ => baseline}/baseline_GREW.yaml (98%) rename config/{ => baseline}/baseline_OUMVLP.yaml (98%) rename {misc/HID => config/baseline}/baseline_hid.yaml (98%) rename config/{ => gaitgl}/gaitgl.yaml (95%) rename config/{ => gaitgl}/gaitgl_OUMVLP.yaml (96%) rename config/{ => gaitpart}/gaitpart.yaml (100%) rename config/{ => gaitpart}/gaitpart_OUMVLP.yaml (96%) rename config/{ => gaitset}/gaitset.yaml (100%) rename config/{ => gaitset}/gaitset_OUMVLP.yaml (95%) rename {misc/partitions => datasets/CASIA-B}/CASIA-B.json (100%) rename {misc/partitions => datasets/CASIA-B}/CASIA-B_include_005.json (100%) create mode 100644 datasets/CASIA-B/README.md rename {misc/partitions => datasets/GREW}/GREW.json (100%) rename {misc => datasets}/GREW/README.md (76%) rename {misc => datasets/GREW}/rearrange_GREW.py (100%) rename {misc => datasets}/HID/HID.json (100%) rename {misc => datasets}/HID/README.md (100%) rename {misc => datasets}/HID/pretreatment_HID.py (100%) rename {misc/partitions => datasets/OUMVLP}/OUMVLP.json (100%) create mode 100644 datasets/OUMVLP/README.md rename {misc => datasets/OUMVLP}/extractor.py (100%) rename {misc => datasets/OUMVLP}/rearrange_OUMVLP.py (100%) create mode 100644 datasets/README.md rename {misc => datasets}/pretreatment.py (100%) diff --git a/config/baseline.yaml b/config/baseline/baseline.yaml similarity index 97% rename from config/baseline.yaml rename to config/baseline/baseline.yaml index ec2c9c8..415bd8a 100644 --- a/config/baseline.yaml +++ b/config/baseline/baseline.yaml @@ -1,7 +1,7 @@ data_cfg: dataset_name: CASIA-B dataset_root: your_path - dataset_partition: ./misc/partitions/CASIA-B_include_005.json + dataset_partition: ./datasets/CASIA-B/CASIA-B_include_005.json num_workers: 1 remove_no_gallery: false # Remove probe if no gallery for it test_dataset_name: CASIA-B diff --git a/config/baseline_GREW.yaml b/config/baseline/baseline_GREW.yaml similarity index 98% rename from config/baseline_GREW.yaml rename to config/baseline/baseline_GREW.yaml index 813dd35..ba32afc 100644 --- a/config/baseline_GREW.yaml +++ b/config/baseline/baseline_GREW.yaml @@ -1,7 +1,7 @@ data_cfg: dataset_name: GREW dataset_root: your_path - dataset_partition: ./misc/partitions/GREW.json + dataset_partition: ./datasets/GREW/GREW.json num_workers: 16 remove_no_gallery: false # Remove probe if no gallery for it test_dataset_name: GREW diff --git a/config/baseline_OUMVLP.yaml b/config/baseline/baseline_OUMVLP.yaml similarity index 98% rename from config/baseline_OUMVLP.yaml rename to config/baseline/baseline_OUMVLP.yaml index 6c2a6c3..b53b5dc 100644 --- a/config/baseline_OUMVLP.yaml +++ b/config/baseline/baseline_OUMVLP.yaml @@ -1,7 +1,7 @@ data_cfg: dataset_name: OUMVLP dataset_root: your_path - dataset_partition: ./misc/partitions/OUMVLP.json + dataset_partition: ./datasets/OUMVLP/OUMVLP.json num_workers: 1 remove_no_gallery: false # Remove probe if no gallery for it test_dataset_name: OUMVLP diff --git a/misc/HID/baseline_hid.yaml b/config/baseline/baseline_hid.yaml similarity index 98% rename from misc/HID/baseline_hid.yaml rename to config/baseline/baseline_hid.yaml index 3555174..6735d79 100644 --- a/misc/HID/baseline_hid.yaml +++ b/config/baseline/baseline_hid.yaml @@ -1,7 +1,7 @@ data_cfg: dataset_name: HID dataset_root: your_path - dataset_partition: ./misc/HID/HID.json + dataset_partition: ./datasets/HID/HID.json num_workers: 1 remove_no_gallery: false # Remove probe if no gallery for it evaluator_cfg: diff --git a/config/default.yaml b/config/default.yaml index 29ce250..b5b772d 100644 --- a/config/default.yaml +++ b/config/default.yaml @@ -2,7 +2,7 @@ data_cfg: dataset_name: CASIA-B dataset_root: your_path num_workers: 1 - dataset_partition: ./misc/partitions/CASIA-B.json + dataset_partition: ./datasets/CASIA-B/CASIA-B.json remove_no_gallery: false cache: false test_dataset_name: CASIA-B diff --git a/config/gaitgl.yaml b/config/gaitgl/gaitgl.yaml similarity index 95% rename from config/gaitgl.yaml rename to config/gaitgl/gaitgl.yaml index b3b4a07..31e1ffc 100644 --- a/config/gaitgl.yaml +++ b/config/gaitgl/gaitgl.yaml @@ -2,7 +2,7 @@ data_cfg: dataset_name: CASIA-B dataset_root: your_path - dataset_partition: ./misc/partitions/CASIA-B_include_005.json + dataset_partition: ./datasets/CASIA-B/CASIA-B_include_005.json num_workers: 1 remove_no_gallery: false test_dataset_name: CASIA-B diff --git a/config/gaitgl_OUMVLP.yaml b/config/gaitgl/gaitgl_OUMVLP.yaml similarity index 96% rename from config/gaitgl_OUMVLP.yaml rename to config/gaitgl/gaitgl_OUMVLP.yaml index 8c6adb0..dcc1969 100644 --- a/config/gaitgl_OUMVLP.yaml +++ b/config/gaitgl/gaitgl_OUMVLP.yaml @@ -2,7 +2,7 @@ data_cfg: dataset_name: OUMVLP dataset_root: your_path - dataset_partition: ./misc/partitions/OUMVLP.json + dataset_partition: ./datasets/OUMVLP/OUMVLP.json num_workers: 1 remove_no_gallery: false test_dataset_name: OUMVLP diff --git a/config/gaitpart.yaml b/config/gaitpart/gaitpart.yaml similarity index 100% rename from config/gaitpart.yaml rename to config/gaitpart/gaitpart.yaml diff --git a/config/gaitpart_OUMVLP.yaml b/config/gaitpart/gaitpart_OUMVLP.yaml similarity index 96% rename from config/gaitpart_OUMVLP.yaml rename to config/gaitpart/gaitpart_OUMVLP.yaml index 9209c25..6c5a972 100644 --- a/config/gaitpart_OUMVLP.yaml +++ b/config/gaitpart/gaitpart_OUMVLP.yaml @@ -1,7 +1,7 @@ data_cfg: dataset_name: OUMVLP dataset_root: your_path - dataset_partition: ./misc/partitions/OUMVLP.json + dataset_partition: ./datasets/OUMVLP/OUMVLP.json num_workers: 4 remove_no_gallery: false test_dataset_name: OUMVLP diff --git a/config/gaitset.yaml b/config/gaitset/gaitset.yaml similarity index 100% rename from config/gaitset.yaml rename to config/gaitset/gaitset.yaml diff --git a/config/gaitset_OUMVLP.yaml b/config/gaitset/gaitset_OUMVLP.yaml similarity index 95% rename from config/gaitset_OUMVLP.yaml rename to config/gaitset/gaitset_OUMVLP.yaml index f79e725..79137c0 100644 --- a/config/gaitset_OUMVLP.yaml +++ b/config/gaitset/gaitset_OUMVLP.yaml @@ -1,7 +1,7 @@ data_cfg: dataset_name: OUMVLP dataset_root: your_path - dataset_partition: ./misc/partitions/OUMVLP.json + dataset_partition: ./datasets/OUMVLP/OUMVLP.json num_workers: 4 remove_no_gallery: false test_dataset_name: OUMVLP diff --git a/config/gln/gln_phase1.yaml b/config/gln/gln_phase1.yaml index 3e3c37e..e2e81f5 100644 --- a/config/gln/gln_phase1.yaml +++ b/config/gln/gln_phase1.yaml @@ -1,7 +1,7 @@ data_cfg: dataset_name: CASIA-B dataset_root: your_path - dataset_partition: ./misc/partitions/CASIA-B_include_005.json + dataset_partition: ./datasets/CASIA-B/CASIA-B_include_005.json num_workers: 1 cache: false remove_no_gallery: false diff --git a/config/gln/gln_phase2.yaml b/config/gln/gln_phase2.yaml index 3d9432c..0b784c5 100644 --- a/config/gln/gln_phase2.yaml +++ b/config/gln/gln_phase2.yaml @@ -1,7 +1,7 @@ data_cfg: dataset_name: CASIA-B dataset_root: your_path - dataset_partition: ./misc/partitions/CASIA-B_include_005.json + dataset_partition: ./datasets/CASIA-B/CASIA-B_include_005.json num_workers: 1 remove_no_gallery: false test_dataset_name: CASIA-B diff --git a/misc/partitions/CASIA-B.json b/datasets/CASIA-B/CASIA-B.json similarity index 100% rename from misc/partitions/CASIA-B.json rename to datasets/CASIA-B/CASIA-B.json diff --git a/misc/partitions/CASIA-B_include_005.json b/datasets/CASIA-B/CASIA-B_include_005.json similarity index 100% rename from misc/partitions/CASIA-B_include_005.json rename to datasets/CASIA-B/CASIA-B_include_005.json diff --git a/datasets/CASIA-B/README.md b/datasets/CASIA-B/README.md new file mode 100644 index 0000000..b6366ec --- /dev/null +++ b/datasets/CASIA-B/README.md @@ -0,0 +1,27 @@ +# CASIA-B +Download URL: http://www.cbsr.ia.ac.cn/GaitDatasetB-silh.zip +- Original + ``` + CASIA-B + 001 (subject) + bg-01 (type) + 000 (view) + 001-bg-01-000-001.png (frame) + 001-bg-01-000-002.png (frame) + ...... + ...... + ...... + ...... + ``` +- Run `python misc/pretreatment.py --input_path CASIA-B --output_path CASIA-B-pkl` +- Processed + ``` + CASIA-B-pkl + 001 (subject) + bg-01 (type) + 000 (view) + 000.pkl (contains all frames) + ...... + ...... + ...... + ``` \ No newline at end of file diff --git a/misc/partitions/GREW.json b/datasets/GREW/GREW.json similarity index 100% rename from misc/partitions/GREW.json rename to datasets/GREW/GREW.json diff --git a/misc/GREW/README.md b/datasets/GREW/README.md similarity index 76% rename from misc/GREW/README.md rename to datasets/GREW/README.md index 0ca8999..51584dd 100644 --- a/misc/GREW/README.md +++ b/datasets/GREW/README.md @@ -26,12 +26,12 @@ After unpacking these compressed files, run this command: Step2 : To rearrange directory of GREW dataset, turning to id-type-view structure, Run ``` -python misc/rearrange_GREW.py --input_path Path_of_GREW-raw --output_path Path_of_GREW-rearranged +python datasets/GREW/rearrange_GREW.py --input_path Path_of_GREW-raw --output_path Path_of_GREW-rearranged ``` Step3: Transforming images to pickle file, run ``` -python misc/pretreatment.py --input_path Path_of_GREW-rearranged --output_path Path_of_GREW-pkl +python datasets/pretreatment.py --input_path Path_of_GREW-rearranged --output_path Path_of_GREW-pkl ``` Then you will see the structure like: @@ -62,17 +62,16 @@ Then you will see the structure like: ``` ## Train the dataset -Modify the `dataset_root` in `./config/baseline_GREW.yaml`, and then run this command: +Modify the `dataset_root` in `./config/baseline/baseline_GREW.yaml`, and then run this command: ```shell -CUDA_VISIBLE_DEVICES=0,1,2,3 python -m torch.distributed.launch --nproc_per_node=4 opengait/main.py --cfgs ./config/baseline_GREW.yaml --phase train +CUDA_VISIBLE_DEVICES=0,1,2,3 python -m torch.distributed.launch --nproc_per_node=4 opengait/main.py --cfgs ./config/baseline/baseline_GREW.yaml --phase train ``` - ## Get the submission file ```shell -CUDA_VISIBLE_DEVICES=0,1,2,3 python -m torch.distributed.launch --nproc_per_node=4 opengait/main.py --cfgs ./config/baseline_GREW.yaml --phase test +CUDA_VISIBLE_DEVICES=0,1,2,3 python -m torch.distributed.launch --nproc_per_node=4 opengait/main.py --cfgs ./config/baseline/baseline_GREW.yaml --phase test ``` The result will be generated in your working directory, you must rename and compress it as the requirements before submitting. ## Evaluation locally -While the original grew treat both seq_01 and seq_02 as gallery, but there is no ground truth for probe. Therefore, it is nessesary to upload the submission file on grew competitation. We seperate test set to: seq_01 as gallery, seq_02 as probe. Then you can modify `eval_func` in the `./config/baseline_GREW.yaml` to `identification_real_scene`, you can obtain result localy like setting of OUMVLP. +While the original grew treat both seq_01 and seq_02 as gallery, but there is no ground truth for probe. Therefore, it is nessesary to upload the submission file on grew competitation. We seperate test set to: seq_01 as gallery, seq_02 as probe. Then you can modify `eval_func` in the `./config/baseline/baseline_GREW.yaml` to `identification_real_scene`, you can obtain result localy like setting of OUMVLP. diff --git a/misc/rearrange_GREW.py b/datasets/GREW/rearrange_GREW.py similarity index 100% rename from misc/rearrange_GREW.py rename to datasets/GREW/rearrange_GREW.py diff --git a/misc/HID/HID.json b/datasets/HID/HID.json similarity index 100% rename from misc/HID/HID.json rename to datasets/HID/HID.json diff --git a/misc/HID/README.md b/datasets/HID/README.md similarity index 100% rename from misc/HID/README.md rename to datasets/HID/README.md diff --git a/misc/HID/pretreatment_HID.py b/datasets/HID/pretreatment_HID.py similarity index 100% rename from misc/HID/pretreatment_HID.py rename to datasets/HID/pretreatment_HID.py diff --git a/misc/partitions/OUMVLP.json b/datasets/OUMVLP/OUMVLP.json similarity index 100% rename from misc/partitions/OUMVLP.json rename to datasets/OUMVLP/OUMVLP.json diff --git a/datasets/OUMVLP/README.md b/datasets/OUMVLP/README.md new file mode 100644 index 0000000..932fa04 --- /dev/null +++ b/datasets/OUMVLP/README.md @@ -0,0 +1,67 @@ +# OUMVLP +Step1: Download URL: http://www.am.sanken.osaka-u.ac.jp/BiometricDB/GaitMVLP.html + +Step2: Unzip the dataset, you will get a structure directory like: +``` +python datasets/OUMVLP/extractor.py --input_path Path_of_OUMVLP-base --output_path Path_of_OUMVLP-raw --password Given_Password +``` + +- Original + ``` + OUMVLP-raw + Silhouette_000-00 (view-sequence) + 00001 (subject) + 0001.png (frame) + 0002.png (frame) + ...... + 00002 + 0001.png (frame) + 0002.png (frame) + ...... + ...... + Silhouette_000-01 + 00001 + 0001.png (frame) + 0002.png (frame) + ...... + 00002 + 0001.png (frame) + 0002.png (frame) + ...... + ...... + Silhouette_015-00 + ...... + Silhouette_015-01 + ...... + ...... + ``` +Step3 : To rearrange directory of OUMVLP dataset, turning to id-type-view structure, Run +``` +python datasets/OUMVLP/rearrange_OUMVLP.py --input_path Path_of_OUMVLP-raw --output_path Path_of_OUMVLP-rearranged +``` + +Step4: Transforming images to pickle file, run +``` +python datasets/pretreatment.py --input_path Path_of_OUMVLP-rearranged --output_path Path_of_OUMVLP-pkl +``` + +- Processed + ``` + OUMVLP-pkl + 00001 (subject) + 00 (sequence) + 000 (view) + 000.pkl (contains all frames) + 015 (view) + 015.pkl (contains all frames) + ... + 01 (sequence) + 000 (view) + 000.pkl (contains all frames) + 015 (view) + 015.pkl (contains all frames) + ...... + 00002 (subject) + ...... + ...... + ``` diff --git a/misc/extractor.py b/datasets/OUMVLP/extractor.py similarity index 100% rename from misc/extractor.py rename to datasets/OUMVLP/extractor.py diff --git a/misc/rearrange_OUMVLP.py b/datasets/OUMVLP/rearrange_OUMVLP.py similarity index 100% rename from misc/rearrange_OUMVLP.py rename to datasets/OUMVLP/rearrange_OUMVLP.py diff --git a/datasets/README.md b/datasets/README.md new file mode 100644 index 0000000..21aa806 --- /dev/null +++ b/datasets/README.md @@ -0,0 +1,24 @@ +# Datasets +OpenGait officially supports a few gait datasets. In order to use them, you need to download them and use the code provided here to pre-process them to the format required by OpenGait. + +## Pre-process +In general, we read the original image provided by the dataset and save a sequence as a pickle file to speed up the training IO. + +The expected dataset structure is as follows: +``` + DATASET_ROOT/ + 001 (subject)/ + bg-01 (type)/ + 000 (view)/ + 000.pkl (contains all frames) + ...... + ...... + ...... +``` + +The specific preprocessing steps are described inside each dataset folder. + +## Split dataset +For each dataset, we split the dataset into training and testing sets. The training set is used to train the model, and the testing set is used to evaluate the model. + +You can use the partition file in [dataset folder](CASIA-B/CASIA-B.json) directly, or you can create yours. Remember to set your path to the partition file in [config/*.yaml](../config/). diff --git a/misc/pretreatment.py b/datasets/pretreatment.py similarity index 100% rename from misc/pretreatment.py rename to datasets/pretreatment.py diff --git a/docs/0.prepare_dataset.md b/docs/0.prepare_dataset.md index 2f468cb..9110c53 100644 --- a/docs/0.prepare_dataset.md +++ b/docs/0.prepare_dataset.md @@ -18,7 +18,7 @@ Download URL: http://www.cbsr.ia.ac.cn/GaitDatasetB-silh.zip ...... ...... ``` -- Run `python misc/pretreatment.py --input_path CASIA-B --output_path CASIA-B-pkl` +- Run `python datasets/pretreatment.py --input_path CASIA-B --output_path CASIA-B-pkl` - Processed ``` CASIA-B-pkl @@ -36,7 +36,7 @@ Step1: Download URL: http://www.am.sanken.osaka-u.ac.jp/BiometricDB/GaitMVLP.htm Step2: Unzip the dataset, you will get a structure directory like: ``` -python misc/extractor.py --input_path Path_of_OUMVLP-base --output_path Path_of_OUMVLP-raw --password Given_Password +python datasets/OUMVLP/extractor.py --input_path Path_of_OUMVLP-base --output_path Path_of_OUMVLP-raw --password Given_Password ``` - Original @@ -70,12 +70,12 @@ python misc/extractor.py --input_path Path_of_OUMVLP-base --output_path Path_of_ ``` Step3 : To rearrange directory of OUMVLP dataset, turning to id-type-view structure, Run ``` -python misc/rearrange_OUMVLP.py --input_path Path_of_OUMVLP-raw --output_path Path_of_OUMVLP-rearranged +python datasets/OUMVLP/rearrange_OUMVLP.py --input_path Path_of_OUMVLP-raw --output_path Path_of_OUMVLP-rearranged ``` Step4: Transforming images to pickle file, run ``` -python misc/pretreatment.py --input_path Path_of_OUMVLP-rearranged --output_path Path_of_OUMVLP-pkl +python datasets/pretreatment.py --input_path Path_of_OUMVLP-rearranged --output_path Path_of_OUMVLP-pkl ``` - Processed @@ -135,12 +135,12 @@ Step2: [Unzip](https://github.com/GREW-Benchmark/GREW-Benchmark) the dataset, yo Step3 : To rearrange directory of GREW dataset, turning to id-type-view structure, Run ``` -python misc/rearrange_GREW.py --input_path Path_of_GREW-raw --output_path Path_of_GREW-rearranged +python datasets/GREW/rearrange_GREW.py --input_path Path_of_GREW-raw --output_path Path_of_GREW-rearranged ``` Step4: Transforming images to pickle file, run ``` -python misc/pretreatment.py --input_path Path_of_GREW-rearranged --output_path Path_of_GREW-pkl +python datasets/pretreatment.py --input_path Path_of_GREW-rearranged --output_path Path_of_GREW-pkl ``` - Processed @@ -170,4 +170,4 @@ python misc/pretreatment.py --input_path Path_of_GREW-rearranged --output_path P ``` ## Split dataset -You can use the partition file in [misc/partitions](misc/partitions/) directly, or you can create yours. Remember to set your path to the partition file in [config/*.yaml](config/). \ No newline at end of file +You can use the partition file in dataset folder directly, or you can create yours. Remember to set your path to the partition file in [config/*.yaml](config/). diff --git a/docs/3.advanced_usages.md b/docs/3.advanced_usages.md index ffe6ea9..1c77818 100644 --- a/docs/3.advanced_usages.md +++ b/docs/3.advanced_usages.md @@ -1,13 +1,13 @@ # Advanced Usages ### Cross-Dataset Evalution -> You can conduct cross-dataset evalution by just modifying several arguments in your [data_cfg](../config/baseline.yaml#L1). +> You can conduct cross-dataset evalution by just modifying several arguments in your [data_cfg](../config/baseline/baseline.yaml#L1). > -> Take [baseline.yaml](../config/baseline.yaml) as an example: +> Take [baseline.yaml](../config/baseline/baseline.yaml) as an example: > ```yaml > data_cfg: > dataset_name: CASIA-B > dataset_root: your_path -> dataset_partition: ./misc/partitions/CASIA-B_include_005.json +> dataset_partition: ./datasets/CASIA-B/CASIA-B_include_005.json > num_workers: 1 > remove_no_gallery: false # Remove probe if no gallery for it > test_dataset_name: CASIA-B @@ -19,7 +19,7 @@ > data_cfg: > dataset_name: CASIA-B > dataset_root: your_OUMVLP_path -> dataset_partition: ./misc/partitions/OUMVLP.json +> dataset_partition: ./datasets/OUMVLP/OUMVLP.json > num_workers: 1 > remove_no_gallery: false # Remove probe if no gallery for it > test_dataset_name: OUMVLP diff --git a/test.sh b/test.sh index 3ed516b..f7656ff 100644 --- a/test.sh +++ b/test.sh @@ -1,15 +1,15 @@ # # **************** For CASIA-B **************** # # Baseline -CUDA_VISIBLE_DEVICES=0,1 python -m torch.distributed.launch --nproc_per_node=2 opengait/main.py --cfgs ./config/baseline.yaml --phase test +CUDA_VISIBLE_DEVICES=0,1 python -m torch.distributed.launch --nproc_per_node=2 opengait/main.py --cfgs ./config/baseline/baseline.yaml --phase test # # GaitSet -# CUDA_VISIBLE_DEVICES=0,1 python -m torch.distributed.launch --nproc_per_node=2 opengait/main.py --cfgs ./config/gaitset.yaml --phase test +# CUDA_VISIBLE_DEVICES=0,1 python -m torch.distributed.launch --nproc_per_node=2 opengait/main.py --cfgs ./config/gaitset/gaitset.yaml --phase test # # GaitPart -# CUDA_VISIBLE_DEVICES=0,1 python -m torch.distributed.launch --nproc_per_node=2 opengait/main.py --cfgs ./config/gaitpart.yaml --phase test +# CUDA_VISIBLE_DEVICES=0,1 python -m torch.distributed.launch --nproc_per_node=2 opengait/main.py --cfgs ./config/gaitpart/gaitpart.yaml --phase test # GaitGL -# CUDA_VISIBLE_DEVICES=0,1,2,3 python -m torch.distributed.launch --master_port 12345 --nproc_per_node=4 opengait/main.py --cfgs ./config/gaitgl.yaml --phase test +# CUDA_VISIBLE_DEVICES=0,1,2,3 python -m torch.distributed.launch --master_port 12345 --nproc_per_node=4 opengait/main.py --cfgs ./config/gaitgl/gaitgl.yaml --phase test # # GLN # # Phase 1 @@ -20,13 +20,13 @@ CUDA_VISIBLE_DEVICES=0,1 python -m torch.distributed.launch --nproc_per_node=2 o # # **************** For OUMVLP **************** # # Baseline -# CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python -m torch.distributed.launch --nproc_per_node=8 opengait/main.py --cfgs ./config/baseline_OUMVLP.yaml --phase test +# CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python -m torch.distributed.launch --nproc_per_node=8 opengait/main.py --cfgs ./config/baseline/baseline_OUMVLP.yaml --phase test # # GaitSet -# CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python -m torch.distributed.launch --nproc_per_node=8 opengait/main.py --cfgs ./config/gaitset_OUMVLP.yaml --phase test +# CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python -m torch.distributed.launch --nproc_per_node=8 opengait/main.py --cfgs ./config/gaitset/gaitset_OUMVLP.yaml --phase test # # GaitPart -# CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python -m torch.distributed.launch --nproc_per_node=8 opengait/main.py --cfgs ./config/gaitpart_OUMVLP.yaml --phase test +# CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python -m torch.distributed.launch --nproc_per_node=8 opengait/main.py --cfgs ./config/gaitpart/gaitpart_OUMVLP.yaml --phase test # GaitGL -# CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python -m torch.distributed.launch --nproc_per_node=8 opengait/main.py --cfgs ./config/gaitgl_OUMVLP.yaml --phase test +# CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python -m torch.distributed.launch --nproc_per_node=8 opengait/main.py --cfgs ./config/gaitgl/gaitgl_OUMVLP.yaml --phase test diff --git a/train.sh b/train.sh index b41e9d4..4dbc531 100644 --- a/train.sh +++ b/train.sh @@ -1,15 +1,15 @@ # # **************** For CASIA-B **************** # # Baseline -CUDA_VISIBLE_DEVICES=0,1 python -m torch.distributed.launch --nproc_per_node=2 opengait/main.py --cfgs ./config/baseline.yaml --phase train +CUDA_VISIBLE_DEVICES=0,1 python -m torch.distributed.launch --nproc_per_node=2 opengait/main.py --cfgs ./config/baseline/baseline.yaml --phase train # # GaitSet -# CUDA_VISIBLE_DEVICES=0,1 python -m torch.distributed.launch --nproc_per_node=2 opengait/main.py --cfgs ./config/gaitset.yaml --phase train +# CUDA_VISIBLE_DEVICES=0,1 python -m torch.distributed.launch --nproc_per_node=2 opengait/main.py --cfgs ./config/gaitset/gaitset.yaml --phase train # # GaitPart -# CUDA_VISIBLE_DEVICES=0,1 python -m torch.distributed.launch --nproc_per_node=2 opengait/main.py --cfgs ./config/gaitpart.yaml --phase train +# CUDA_VISIBLE_DEVICES=0,1 python -m torch.distributed.launch --nproc_per_node=2 opengait/main.py --cfgs ./config/gaitpart/gaitpart.yaml --phase train # GaitGL -# CUDA_VISIBLE_DEVICES=0,1,2,3 python -m torch.distributed.launch --nproc_per_node=4 opengait/main.py --cfgs ./config/gaitgl.yaml --phase train +# CUDA_VISIBLE_DEVICES=0,1,2,3 python -m torch.distributed.launch --nproc_per_node=4 opengait/main.py --cfgs ./config/gaitgl/gaitgl.yaml --phase train # # GLN # # Phase 1 @@ -20,13 +20,13 @@ CUDA_VISIBLE_DEVICES=0,1 python -m torch.distributed.launch --nproc_per_node=2 o # # **************** For OUMVLP **************** # # Baseline -# CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python -m torch.distributed.launch --nproc_per_node=8 opengait/main.py --cfgs ./config/baseline_OUMVLP.yaml --phase train +# CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python -m torch.distributed.launch --nproc_per_node=8 opengait/main.py --cfgs ./config/baseline/baseline_OUMVLP.yaml --phase train # # GaitSet -# CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python -m torch.distributed.launch --nproc_per_node=8 opengait/main.py --cfgs ./config/gaitset_OUMVLP.yaml --phase train +# CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python -m torch.distributed.launch --nproc_per_node=8 opengait/main.py --cfgs ./config/gaitset/gaitset_OUMVLP.yaml --phase train # # GaitPart -# CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python -m torch.distributed.launch --nproc_per_node=8 opengait/main.py --cfgs ./config/gaitpart_OUMVLP.yaml --phase train +# CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python -m torch.distributed.launch --nproc_per_node=8 opengait/main.py --cfgs ./config/gaitpart/gaitpart_OUMVLP.yaml --phase train # GaitGL -# CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python -m torch.distributed.launch --nproc_per_node=8 opengait/main.py --cfgs ./config/gaitgl_OUMVLP.yaml --phase train \ No newline at end of file +# CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python -m torch.distributed.launch --nproc_per_node=8 opengait/main.py --cfgs ./config/gaitgl/gaitgl_OUMVLP.yaml --phase train