Refine DRF preprocessing and body-prior pipeline

This commit is contained in:
2026-03-08 04:04:15 +08:00
parent fddbf6eeda
commit bbb41e8dd9
10 changed files with 448 additions and 53 deletions
+5 -5
View File
@@ -1,6 +1,6 @@
data_cfg:
dataset_name: Scoliosis1K
dataset_root: /mnt/public/data/Scoliosis1K/Scoliosis1K-drf-pkl-118
dataset_root: /mnt/public/data/Scoliosis1K/Scoliosis1K-drf-pkl-118-paper
dataset_partition: ./datasets/Scoliosis1K/Scoliosis1K_118.json
num_workers: 1
remove_no_gallery: false
@@ -10,7 +10,7 @@ evaluator_cfg:
enable_float16: true
restore_ckpt_strict: true
restore_hint: 20000
save_name: DRF
save_name: DRF_paper
eval_func: evaluate_scoliosis
sampler:
batch_shuffle: false
@@ -19,7 +19,7 @@ evaluator_cfg:
frames_all_limit: 720
metric: euc
transform:
- type: BaseSilCuttingTransform
- type: BaseSilTransform
- type: NoOperation
loss_cfg:
@@ -90,7 +90,7 @@ trainer_cfg:
restore_ckpt_strict: true
restore_hint: 0
save_iter: 20000
save_name: DRF
save_name: DRF_paper
sync_BN: true
total_iter: 20000
sampler:
@@ -102,5 +102,5 @@ trainer_cfg:
sample_type: fixed_unordered
type: TripletSampler
transform:
- type: BaseSilCuttingTransform
- type: BaseSilTransform
- type: NoOperation