Align DRF skeleton preprocessing with upstream heatmap path

2026-03-08 14:50:35 +08:00
parent bbb41e8dd9
commit 295d951206
10 changed files with 174 additions and 21 deletions
@@ -1,6 +1,6 @@
 data_cfg:
  dataset_name: Scoliosis1K
-  dataset_root: /mnt/public/data/Scoliosis1K/Scoliosis1K-drf-pkl-118-paper
+  dataset_root: /mnt/public/data/Scoliosis1K/Scoliosis1K-drf-pkl-118-aligned
  dataset_partition: ./datasets/Scoliosis1K/Scoliosis1K_118.json
  num_workers: 1
  remove_no_gallery: false
@@ -19,7 +19,7 @@ evaluator_cfg:
    frames_all_limit: 720
  metric: euc
  transform:
-    - type: BaseSilTransform
+    - type: BaseSilCuttingTransform
    - type: NoOperation

 loss_cfg:
@@ -102,5 +102,5 @@ trainer_cfg:
    sample_type: fixed_unordered
    type: TripletSampler
  transform:
-    - type: BaseSilTransform
+    - type: BaseSilCuttingTransform
    - type: NoOperation