Commit ·
5a65b79
0
Parent(s):
SONIC G1 cartwheel ONNX policy (final, root-refined)
Browse files- .gitattributes +1 -0
- README.md +46 -0
- model_step_001500_decoder.onnx +3 -0
- model_step_001500_encoder.onnx +3 -0
- model_step_001500_g1.onnx +3 -0
- model_step_001500_smpl.onnx +3 -0
- model_step_001500_teleop.onnx +3 -0
.gitattributes
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
*.onnx filter=lfs diff=lfs merge=lfs -text
|
README.md
ADDED
|
@@ -0,0 +1,46 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: other
|
| 3 |
+
tags:
|
| 4 |
+
- robotics
|
| 5 |
+
- unitree-g1
|
| 6 |
+
- cartwheel
|
| 7 |
+
- onnx
|
| 8 |
+
- isaac-lab
|
| 9 |
+
---
|
| 10 |
+
|
| 11 |
+
# SONIC G1 Cartwheel — ONNX Policy
|
| 12 |
+
|
| 13 |
+
Exported ONNX policy for the G1 cartwheel motion-imitation agent (GEAR-SONIC /
|
| 14 |
+
SONIC fine-tune). Exported with `+export_onnx_only=true` from the final
|
| 15 |
+
checkpoint (`FT-full-cont-1500-anchor`, 1500 iterations of root-refinement
|
| 16 |
+
fine-tuning; full pipeline in the private GitHub repo
|
| 17 |
+
`qjwdlwjdl/zest-g1-cartwheel-sonic`).
|
| 18 |
+
|
| 19 |
+
## Files
|
| 20 |
+
|
| 21 |
+
| File | Role |
|
| 22 |
+
|---|---|
|
| 23 |
+
| `model_step_001500_teleop.onnx` | teleoperation head |
|
| 24 |
+
| `model_step_001500_encoder.onnx` | observation encoder |
|
| 25 |
+
| `model_step_001500_decoder.onnx` | action decoder / policy head |
|
| 26 |
+
| `model_step_001500_g1.onnx` | G1 model export |
|
| 27 |
+
| `model_step_001500_smpl.onnx` | SMPL export |
|
| 28 |
+
|
| 29 |
+
## Evaluation (full 7.06 s / 352-step window, 32 rollouts)
|
| 30 |
+
|
| 31 |
+
- mpjpe_g 101.5 mm · mpjpe_l 41.3 mm · root drift 90.5 mm
|
| 32 |
+
- Best rollout: 58 mm mean over the full window, 33 mm final-frame error,
|
| 33 |
+
pelvis 0.79 m at the end (reference 0.784 m) — completes the cartwheel and
|
| 34 |
+
returns to standing.
|
| 35 |
+
- Stock `sonic_release` baseline under the same protocol: 913.7 mm.
|
| 36 |
+
|
| 37 |
+
Renders: before/after videos in the linked GitHub repo — the stock model
|
| 38 |
+
collapses at the start of the move; the fine-tuned model completes the move and
|
| 39 |
+
lands standing.
|
| 40 |
+
|
| 41 |
+
## Attribution
|
| 42 |
+
|
| 43 |
+
- Base weights: `nvidia/GEAR-SONIC` (NVIDIA, GEAR-SONIC / SONIC).
|
| 44 |
+
- Task reference: ZEST (`10.5281/zenodo.21135719`, CC-BY-4.0).
|
| 45 |
+
- Simulation: Isaac Lab v2.3.2 / Isaac Sim 5.1.0.
|
| 46 |
+
- Motion Data by Bones Studio (as required when publicly posting).
|
model_step_001500_decoder.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:db7ac8c5aa900ae62b337a9d4ffab767fd04ccc27018a735854d2bf180d5005a
|
| 3 |
+
size 40868426
|
model_step_001500_encoder.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4b5ac67e1e009a17c43efd4cc90640368169171aae8ad07368903cc9fe7947d6
|
| 3 |
+
size 49949761
|
model_step_001500_g1.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0c51c1003c17eeced549e7f71cc9438047767b44ef5c2576256e2c1527a709f6
|
| 3 |
+
size 57947822
|
model_step_001500_smpl.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:58ff7ab087430ec01bfb07bf4d3b7bbfaa768b9c839ed65512955c3a97fdf2c9
|
| 3 |
+
size 59600909
|
model_step_001500_teleop.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:501b717f8ade56c012b228b78b66546dcb5bc0f0acc6171c53781b8381177be4
|
| 3 |
+
size 54916394
|