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: data_cfg:
dataset_name: Scoliosis1K dataset_name: Scoliosis1K
dataset_root: /mnt/public/data/Scoliosis1K/Scoliosis1K-drf-pkl dataset_root: /mnt/public/data/Scoliosis1K/Scoliosis1K-drf-pkl-118
dataset_partition: ./datasets/Scoliosis1K/Scoliosis1K_1116.json dataset_partition: ./datasets/Scoliosis1K/Scoliosis1K_118.json
num_workers: 1 num_workers: 1
remove_no_gallery: false remove_no_gallery: false
test_dataset_name: Scoliosis1K test_dataset_name: Scoliosis1K
+2 -2
View File
@@ -1,7 +1,7 @@
data_cfg: data_cfg:
dataset_name: Scoliosis1K dataset_name: Scoliosis1K
dataset_root: /mnt/public/data/Scoliosis1K/Scoliosis1K-drf-pkl dataset_root: /mnt/public/data/Scoliosis1K/Scoliosis1K-drf-pkl-118
dataset_partition: ./datasets/Scoliosis1K/Scoliosis1K_1116.json dataset_partition: ./datasets/Scoliosis1K/Scoliosis1K_118.json
num_workers: 1 num_workers: 1
remove_no_gallery: false remove_no_gallery: false
test_dataset_name: Scoliosis1K test_dataset_name: Scoliosis1K
+1 -1
View File
@@ -107,7 +107,7 @@ Run:
uv run python datasets/pretreatment_scoliosis_drf.py \ uv run python datasets/pretreatment_scoliosis_drf.py \
--pose_data_path=<path_to_pose_pkl> \ --pose_data_path=<path_to_pose_pkl> \
--output_path=<path_to_drf_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: The output layout is: