JeffrinSam commited on
Commit
f8e7d09
·
verified ·
1 Parent(s): c8529e9

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +90 -0
README.md ADDED
@@ -0,0 +1,90 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ language:
4
+ - en
5
+ tags:
6
+ - robotics
7
+ - navigation
8
+ - video-to-navigation
9
+ - diffusion-transformer
10
+ - optical-flow
11
+ - humanoid
12
+ - GENESIS
13
+ - unitree-g1
14
+ library_name: pytorch
15
+ pipeline_tag: robotics
16
+ ---
17
+
18
+ # FlowDiT V3 Humanoid — Video-to-Navigation (GENESIS)
19
+
20
+ Part of the **GENESIS** research framework: video-conditioned robot learning.
21
+
22
+ **Paper**: [Action Agent: Agentic Video Generation Meets Flow-Constrained Diffusion](https://arxiv.org/abs/2605.01477) (IROS 2026)
23
+
24
+ **Code**: [github.com/jeffrinsam/GENESIS](https://github.com/jeffrinsam/GENESIS) → `part2_navigation/flow_constrained_v3_humanoid/`
25
+
26
+ ## Model Description
27
+
28
+ FlowDiT V3 Humanoid is an inference-optimized Diffusion Transformer specialized for **bipedal humanoid navigation** (Unitree G1). It extends FlowDiT V2 with humanoid-specific motion constraints and whole-body balance priors.
29
+
30
+ **Architecture:**
31
+ - **Visual encoder**: DINOv2-ViT-B/14 (frozen)
32
+ - **Flow encoder**: RAFT optical flow with humanoid-specific temporal attention
33
+ - **DiT backbone**: Enlarged Diffusion Transformer with balance-constraint cross-attention
34
+ - **Output**: 3-DOF velocity command `[vx, vy, yaw_rate]` + gait phase signal
35
+
36
+ **Target robot**: Unitree G1 humanoid (inference only — see code for the training pipeline).
37
+
38
+ **Runtime**: PyTorch 2.9.1+cu128, requires ~4 GB VRAM for inference.
39
+
40
+ ## Performance
41
+
42
+ Evaluated on Unitree G1 in Isaac Sim navigation tasks:
43
+
44
+ | Metric | Value |
45
+ |--------|-------|
46
+ | Success Rate (SR @ 3.0 m) | 100% |
47
+ | SR @ 1.0 m (post-processed) | ~39% |
48
+ | Avg Trajectory Error (ATE) | 0.38 m |
49
+
50
+ ## Usage
51
+
52
+ ```bash
53
+ # Activate the V3 inference venv (torch 2.9.1+cu128)
54
+ cd GENESIS/part2_navigation/flow_constrained_v3_humanoid
55
+ source .venv/bin/activate
56
+
57
+ python infer_humanoid.py \
58
+ --checkpoint flowdit_v3_humanoid_best.pt \
59
+ --goal_video goal.mp4 \
60
+ --current_obs obs.jpg
61
+ ```
62
+
63
+ Download via the GENESIS checkpoint script:
64
+ ```bash
65
+ bash scripts/download_checkpoints.sh
66
+ ```
67
+
68
+ ## Checkpoint Details
69
+
70
+ | File | Size | Format |
71
+ |------|------|--------|
72
+ | `flowdit_v3_humanoid_best.pt` | 982 MB | PyTorch state dict + config |
73
+
74
+ ## Citation
75
+
76
+ ```bibtex
77
+ @inproceedings{sam2026actionagent,
78
+ title = {Action Agent: Agentic Video Generation Meets Flow-Constrained Diffusion},
79
+ author = {Sam, Jeffrin and Khang, Nguyen and Mahmoud, Yara and
80
+ Altamirano Cabrera, Miguel and Tsetserukou, Dzmitry},
81
+ booktitle = {2026 IEEE/RSJ International Conference on Intelligent Robots
82
+ and Systems (IROS)},
83
+ year = {2026},
84
+ note = {arXiv:2605.01477}
85
+ }
86
+ ```
87
+
88
+ ## License
89
+
90
+ Apache 2.0. See [LICENSE](https://github.com/jeffrinsam/GENESIS/blob/main/LICENSE).