Document ScoNet naming and Scoliosis eval configs

This commit is contained in:
2026-03-08 17:34:33 +08:00
parent 295d951206
commit 4e0b0a18dc
5 changed files with 315 additions and 74 deletions
@@ -1,10 +1,7 @@
data_cfg:
dataset_name: Scoliosis1K
dataset_root: /mnt/public/data/Scoliosis1K/Scoliosis1K-drf-pkl-118-aligned
dataset_root: /mnt/public/data/Scoliosis1K/Scoliosis1K_sigma_8.0/pkl
dataset_partition: ./datasets/Scoliosis1K/Scoliosis1K_118.json
data_in_use:
- true
- false
num_workers: 1
remove_no_gallery: false
test_dataset_name: Scoliosis1K
@@ -13,7 +10,7 @@ evaluator_cfg:
enable_float16: true
restore_ckpt_strict: true
restore_hint: 20000
save_name: ScoNet_skeleton_118
save_name: ScoNet_skeleton_118_sigma8
eval_func: evaluate_scoliosis
sampler:
batch_shuffle: false
@@ -90,7 +87,7 @@ trainer_cfg:
restore_ckpt_strict: true
restore_hint: 0
save_iter: 20000
save_name: ScoNet_skeleton_118
save_name: ScoNet_skeleton_118_sigma8
sync_BN: true
total_iter: 20000
sampler: