oniichan521 commited on
Commit
5a65b79
·
0 Parent(s):

SONIC G1 cartwheel ONNX policy (final, root-refined)

Browse files
.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