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