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
@@ -109,9 +109,17 @@ uv run python datasets/pretreatment_scoliosis_drf.py \
  --output_path=<path_to_drf_pkl>
 ```

-To reproduce the paper defaults more closely, the script now uses
-`configs/drf/pretreatment_heatmap_drf.yaml` by default, which enables
-summed two-channel skeleton maps and a literal 128-pixel height normalization.
+The script uses `configs/drf/pretreatment_heatmap_drf.yaml` by default.
+That keeps the upstream OpenGait/SkeletonGait heatmap behavior from
+commit `f754f6f3831e9f83bb28f4e2f63dd43d8bcf9dc4` for the skeleton-map
+branch while still building the DRF-specific two-channel output.
+
+If you explicitly want the more paper-literal summed heatmap ablation, add:
+
+```bash
+  --heatmap_reduction=sum
+```
+
 If you explicitly want train-only PAV min-max statistics, add:

 ```bash