Align DRF Scoliosis1K split with paper

This commit is contained in:
2026-03-07 18:03:53 +08:00
parent 51eee70a4b
commit 5c6bef1ca1
3 changed files with 5 additions and 5 deletions
+2 -2
View File
@@ -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
+2 -2
View File
@@ -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
+1 -1
View File
@@ -107,7 +107,7 @@ Run:
uv run python datasets/pretreatment_scoliosis_drf.py \
--pose_data_path=<path_to_pose_pkl> \
--output_path=<path_to_drf_pkl> \
--stats_partition=./datasets/Scoliosis1K/Scoliosis1K_1116.json
--stats_partition=./datasets/Scoliosis1K/Scoliosis1K_118.json
```
The output layout is: