feat: add drf author checkpoint compatibility bundle

2026-03-14 17:12:27 +08:00
parent d4e2a59ad2
commit 5f98844aff
18 changed files with 1144 additions and 8 deletions
@@ -0,0 +1,62 @@
+# DRF Author Checkpoint Artifact
+
+This artifact bundle archives the author-provided DRF checkpoint together with the compatibility configs that make it usable in this repo.
+
+Files:
+- `DRF_118_unordered_iter2w_lr0.001_8830-08000.pt`
+- `author_original_config.yaml`
+- `eval_112_1gpu.yaml`
+- `eval_118_defaultroot_1gpu.yaml`
+- `eval_118_splitroot_1gpu.yaml`
+- `eval_118_aligned_1gpu.yaml`
+- `eval_118_paper_1gpu.yaml`
+
+Best recovered path:
+- config: `configs/drf/drf_author_eval_118_aligned_1gpu.yaml`
+- result:
+  - `80.24 Acc / 76.73 Prec / 76.40 Rec / 76.56 F1`
+
+## Run Commands
+
+Recommended eval:
+
+```bash
+CUDA_VISIBLE_DEVICES=GPU-9cc7b26e-90d4-0c49-4d4c-060e528ffba6 \
+uv run torchrun --nproc_per_node=1 --master_port=29693 \
+  opengait/main.py \
+  --cfgs ./configs/drf/drf_author_eval_118_aligned_1gpu.yaml \
+  --phase test
+```
+
+Other compatibility checks:
+
+```bash
+CUDA_VISIBLE_DEVICES=GPU-9cc7b26e-90d4-0c49-4d4c-060e528ffba6 \
+uv run torchrun --nproc_per_node=1 --master_port=29695 \
+  opengait/main.py \
+  --cfgs ./configs/drf/drf_author_eval_112_1gpu.yaml \
+  --phase test
+
+CUDA_VISIBLE_DEVICES=GPU-9cc7b26e-90d4-0c49-4d4c-060e528ffba6 \
+uv run torchrun --nproc_per_node=1 --master_port=29696 \
+  opengait/main.py \
+  --cfgs ./configs/drf/drf_author_eval_118_splitroot_1gpu.yaml \
+  --phase test
+
+CUDA_VISIBLE_DEVICES=GPU-9cc7b26e-90d4-0c49-4d4c-060e528ffba6 \
+uv run torchrun --nproc_per_node=1 --master_port=29697 \
+  opengait/main.py \
+  --cfgs ./configs/drf/drf_author_eval_118_paper_1gpu.yaml \
+  --phase test
+```
+
+## Paths To Change On Another Machine
+
+If you move this artifact bundle or use a different dataset location, update:
+- `data_cfg.dataset_root`
+- `data_cfg.dataset_partition`
+- `evaluator_cfg.restore_hint`
+
+In this repo, the checkpoint is archived at:
+- `artifact/scoliosis_drf_author_118_compat/DRF_118_unordered_iter2w_lr0.001_8830-08000.pt`
+