diff --git a/configs/baseline/baseline.yaml b/configs/baseline/baseline.yaml index a7aecc1..3fb4b67 100644 --- a/configs/baseline/baseline.yaml +++ b/configs/baseline/baseline.yaml @@ -1,7 +1,7 @@ data_cfg: dataset_name: CASIA-B dataset_root: your_path - dataset_partition: ./datasets/CASIA-B/CASIA-B_include_005.json + dataset_partition: ./datasets/CASIA-B/CASIA-B.json num_workers: 1 remove_no_gallery: false # Remove probe if no gallery for it test_dataset_name: CASIA-B diff --git a/configs/gaitgl/gaitgl.yaml b/configs/gaitgl/gaitgl.yaml index bce9272..8c263ce 100644 --- a/configs/gaitgl/gaitgl.yaml +++ b/configs/gaitgl/gaitgl.yaml @@ -2,7 +2,7 @@ data_cfg: dataset_name: CASIA-B dataset_root: your_path - dataset_partition: ./datasets/CASIA-B/CASIA-B_include_005.json + dataset_partition: ./datasets/CASIA-B/CASIA-B.json num_workers: 1 remove_no_gallery: false test_dataset_name: CASIA-B diff --git a/configs/gln/gln_phase1.yaml b/configs/gln/gln_phase1.yaml index db36c5f..34f8991 100644 --- a/configs/gln/gln_phase1.yaml +++ b/configs/gln/gln_phase1.yaml @@ -1,7 +1,7 @@ data_cfg: dataset_name: CASIA-B dataset_root: your_path - dataset_partition: ./datasets/CASIA-B/CASIA-B_include_005.json + dataset_partition: ./datasets/CASIA-B/CASIA-B.json num_workers: 1 cache: false remove_no_gallery: false diff --git a/configs/gln/gln_phase2.yaml b/configs/gln/gln_phase2.yaml index 0b784c5..260bf27 100644 --- a/configs/gln/gln_phase2.yaml +++ b/configs/gln/gln_phase2.yaml @@ -1,7 +1,7 @@ data_cfg: dataset_name: CASIA-B dataset_root: your_path - dataset_partition: ./datasets/CASIA-B/CASIA-B_include_005.json + dataset_partition: ./datasets/CASIA-B/CASIA-B.json num_workers: 1 remove_no_gallery: false test_dataset_name: CASIA-B diff --git a/datasets/CASIA-B/CASIA-B.json b/datasets/CASIA-B/CASIA-B.json index 04565bc..a4daee6 100644 --- a/datasets/CASIA-B/CASIA-B.json +++ b/datasets/CASIA-B/CASIA-B.json @@ -4,6 +4,7 @@ "002", "003", "004", + "005", "006", "007", "008", diff --git a/datasets/CASIA-B/CASIA-B_include_005.json b/datasets/CASIA-B/CASIA-B_include_005.json deleted file mode 100644 index a4daee6..0000000 --- a/datasets/CASIA-B/CASIA-B_include_005.json +++ /dev/null @@ -1,130 +0,0 @@ -{ - "TRAIN_SET": [ - "001", - "002", - "003", - "004", - "005", - "006", - "007", - "008", - "009", - "010", - "011", - "012", - "013", - "014", - "015", - "016", - "017", - "018", - "019", - "020", - "021", - "022", - "023", - "024", - "025", - "026", - "027", - "028", - "029", - "030", - "031", - "032", - "033", - "034", - "035", - "036", - "037", - "038", - "039", - "040", - "041", - "042", - "043", - "044", - "045", - "046", - "047", - "048", - "049", - "050", - "051", - "052", - "053", - "054", - "055", - "056", - "057", - "058", - "059", - "060", - "061", - "062", - "063", - "064", - "065", - "066", - "067", - "068", - "069", - "070", - "071", - "072", - "073", - "074" - ], - "TEST_SET": [ - "075", - "076", - "077", - "078", - "079", - "080", - "081", - "082", - "083", - "084", - "085", - "086", - "087", - "088", - "089", - "090", - "091", - "092", - "093", - "094", - "095", - "096", - "097", - "098", - "099", - "100", - "101", - "102", - "103", - "104", - "105", - "106", - "107", - "108", - "109", - "110", - "111", - "112", - "113", - "114", - "115", - "116", - "117", - "118", - "119", - "120", - "121", - "122", - "123", - "124" - ] -} \ No newline at end of file diff --git a/docs/3.detailed_config.md b/docs/3.detailed_config.md index 5186caa..53c2ed4 100644 --- a/docs/3.detailed_config.md +++ b/docs/3.detailed_config.md @@ -86,7 +86,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.json num_workers: 1 remove_no_gallery: false # Remove probe if no gallery for it test_dataset_name: CASIA-B @@ -96,7 +96,7 @@ evaluator_cfg: restore_ckpt_strict: true restore_hint: 60000 save_name: Baseline - eval_func: identification + eval_func: evaluate_indoor_dataset sampler: batch_shuffle: false batch_size: 16 diff --git a/docs/5.advanced_usages.md b/docs/5.advanced_usages.md index f212b6e..54adf40 100644 --- a/docs/5.advanced_usages.md +++ b/docs/5.advanced_usages.md @@ -7,7 +7,7 @@ > data_cfg: > dataset_name: CASIA-B > dataset_root: your_path -> dataset_partition: ./datasets/CASIA-B/CASIA-B_include_005.json +> dataset_partition: ./datasets/CASIA-B/CASIA-B.json > num_workers: 1 > remove_no_gallery: false # Remove probe if no gallery for it > test_dataset_name: CASIA-B