Align DRF Scoliosis1K split with paper
This commit is contained in:
@@ -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,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
|
||||
|
||||
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user