JeffrinSam commited on
Commit
6d0b8f4
·
verified ·
1 Parent(s): 01992a3

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +99 -0
README.md ADDED
@@ -0,0 +1,99 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ - GENESIS
12
+ library_name: pytorch
13
+ pipeline_tag: robotics
14
+ ---
15
+
16
+ # FlowDiT V2 — Video-to-Navigation (GENESIS)
17
+
18
+ Part of the **GENESIS** research framework: video-conditioned robot learning.
19
+
20
+ **Paper**: [Action Agent: Agentic Video Generation Meets Flow-Constrained Diffusion](https://arxiv.org/abs/2605.01477) (IROS 2026)
21
+
22
+ **Code**: [github.com/jeffrinsam/GENESIS](https://github.com/jeffrinsam/GENESIS) → `part2_navigation/`
23
+
24
+ ## Model Description
25
+
26
+ FlowDiT V2 is a flow-constrained Diffusion Transformer (DiT) that translates a reference goal video and a current observation image into continuous navigation commands `[vx, vy, yaw_rate]`.
27
+
28
+ **Architecture:**
29
+ - **Visual encoder**: DINOv2-ViT-B/14 (frozen) — extracts spatial features from current observation
30
+ - **Flow encoder**: RAFT optical flow → temporal flow tokens from goal video
31
+ - **DiT backbone**: Diffusion Transformer with cross-attention between flow tokens and obs features
32
+ - **Output**: 3-DOF velocity command at 2 Hz control frequency
33
+
34
+ **Training data**: ~50k episodes across wheeled and legged embodiments (Unitree B1, G1, custom wheeled platforms) in Isaac Sim.
35
+
36
+ ## Performance
37
+
38
+ Evaluated on 41 tasks across 3 robot embodiments in Isaac Sim:
39
+
40
+ | Metric | Value |
41
+ |--------|-------|
42
+ | Success Rate (SR @ 3.0 m) | 100% |
43
+ | SPL | 0.91 |
44
+ | Avg Trajectory Error (ATE) | 0.42 m |
45
+ | Direction Accuracy (cosine > 0.75) | 96.3% |
46
+
47
+ ## Usage
48
+
49
+ ```bash
50
+ # Install dependencies
51
+ conda activate genesis-navigation
52
+ cd GENESIS
53
+
54
+ # Single inference
55
+ python part2_navigation/flow_constrained_v2/inference.py \
56
+ --checkpoint path/to/best.pth \
57
+ --goal_video reference.mp4 \
58
+ --current_obs frame.jpg \
59
+ --output actions.npy
60
+ ```
61
+
62
+ Download via the GENESIS checkpoint script:
63
+ ```bash
64
+ bash scripts/download_checkpoints.sh
65
+ ```
66
+
67
+ ## Checkpoint Details
68
+
69
+ | File | Size | Format |
70
+ |------|------|--------|
71
+ | `best.pth` | 1.2 GB | PyTorch state dict + config |
72
+
73
+ The `.pth` file contains:
74
+ ```python
75
+ {
76
+ "model_state_dict": ...,
77
+ "config": {"use_raft": False, "hidden_dim": 512, ...},
78
+ "epoch": ...,
79
+ "val_loss": ...
80
+ }
81
+ ```
82
+
83
+ ## Citation
84
+
85
+ ```bibtex
86
+ @inproceedings{sam2026actionagent,
87
+ title = {Action Agent: Agentic Video Generation Meets Flow-Constrained Diffusion},
88
+ author = {Sam, Jeffrin and Khang, Nguyen and Mahmoud, Yara and
89
+ Altamirano Cabrera, Miguel and Tsetserukou, Dzmitry},
90
+ booktitle = {2026 IEEE/RSJ International Conference on Intelligent Robots
91
+ and Systems (IROS)},
92
+ year = {2026},
93
+ note = {arXiv:2605.01477}
94
+ }
95
+ ```
96
+
97
+ ## License
98
+
99
+ Apache 2.0. See [LICENSE](https://github.com/jeffrinsam/GENESIS/blob/main/LICENSE).