diff --git a/configs/drf/drf_scoliosis1k.yaml b/configs/drf/drf_scoliosis1k.yaml index 1307744..221f75f 100644 --- a/configs/drf/drf_scoliosis1k.yaml +++ b/configs/drf/drf_scoliosis1k.yaml @@ -1,7 +1,7 @@ data_cfg: dataset_name: Scoliosis1K - dataset_root: /mnt/public/data/Scoliosis1K/Scoliosis1K-drf-pkl - dataset_partition: ./datasets/Scoliosis1K/Scoliosis1K_1116.json + dataset_root: /mnt/public/data/Scoliosis1K/Scoliosis1K-drf-pkl-118 + dataset_partition: ./datasets/Scoliosis1K/Scoliosis1K_118.json num_workers: 1 remove_no_gallery: false test_dataset_name: Scoliosis1K diff --git a/configs/drf/drf_scoliosis1k_smoke.yaml b/configs/drf/drf_scoliosis1k_smoke.yaml index a10bfd8..9aeae2f 100644 --- a/configs/drf/drf_scoliosis1k_smoke.yaml +++ b/configs/drf/drf_scoliosis1k_smoke.yaml @@ -1,7 +1,7 @@ data_cfg: dataset_name: Scoliosis1K - dataset_root: /mnt/public/data/Scoliosis1K/Scoliosis1K-drf-pkl - dataset_partition: ./datasets/Scoliosis1K/Scoliosis1K_1116.json + dataset_root: /mnt/public/data/Scoliosis1K/Scoliosis1K-drf-pkl-118 + dataset_partition: ./datasets/Scoliosis1K/Scoliosis1K_118.json num_workers: 1 remove_no_gallery: false test_dataset_name: Scoliosis1K diff --git a/datasets/Scoliosis1K/README.md b/datasets/Scoliosis1K/README.md index 3b89bf1..8e22845 100644 --- a/datasets/Scoliosis1K/README.md +++ b/datasets/Scoliosis1K/README.md @@ -107,7 +107,7 @@ Run: uv run python datasets/pretreatment_scoliosis_drf.py \ --pose_data_path= \ --output_path= \ - --stats_partition=./datasets/Scoliosis1K/Scoliosis1K_1116.json + --stats_partition=./datasets/Scoliosis1K/Scoliosis1K_118.json ``` The output layout is: