ARTICLE · INTELLIGENCE

战地情报 · 详情页

来自尧图项目组的一线实战观察与深度解析

pi0代码解析-examples/aloha_real-4

pi0代码解析-examples/aloha_real-4

openpi(π₀ / π₀.₅ / π₀-FAST 机器人策略模型)在 ALOHA 真机平台上的完整示例工程,覆盖“训练数据准备 → 真机推理部署 → 数据采集辅助”三个环节。

它把ALOHA 双臂机器人真机接入 openpi 的远程 WebSocket 策略服务,实现“观测采集 → 网络推理 → 动作执行”的闭环,并提供数据格式转换与底层硬件工具。

1.main.py—— 真机推理入口

整个文件的最终目的:让真机跑起来。

WebsocketClientPolicy (远程调用策略服务器) ↓ 包装 ActionChunkBroker (一次推理、H 步复用,H=25) ↓ 包装 PolicyAgent (适配 runtime 的 Agent 接口) ↓ 注入 Runtime(environment=真机, max_hz=50, num_episodes, max_episode_steps)

关键设计:

  • 从服务端元数据取reset_pose,保证每个 episode 的初始位姿与策略训练时一致;
  • max_episode_steps是安全兜底,因为真机没有自动终止信号;
  • 动作块
RELATED READING

延伸阅读

更多一线实战笔记与深度复盘,助您持续精进