ACT++ 模仿学习算法库

act-plus-plus

比赛/竞技 高级 🧩 软硬件结合 已发布
MarkFzp 3622 Stars MIT BOM 完整度: 3/5 教程完整度: 0/5

项目简介

Co-training模仿学习算法用于Mobile ALOHA:ACT、Diffusion Policy、VINN。


act-plus-plus是一个专注于模仿学习算法与协同训练的开源项目,主要面向Mobile ALOHA机器人平台。该项目实现了三种核心算法:ACT(Action Chunking with Transformers)、Diffusion Policy和VINN,并提供了两个仿真环境——Transfer Cube和Bimanual Insertion,支持用户在仿真或真实机器人上进行训练和评估。

标签

项目特点

实现三种主流模仿学习算法:ACT、Diffusion Policy 和 VINN
提供两个模拟环境:Transfer Cube(转移立方体)和 Bimanual Insertion(双臂插入)
支持模拟环境和真实机器人部署
提供脚本化策略生成演示数据
支持时间集成(temporal ensembling)提升策略平滑度
包含完整的训练、评估和数据可视化工具
与 Mobile ALOHA 硬件系统无缝集成

技术规格

支持算法
模拟环境
控制方式
框架依赖
Python版本
数据格式
硬件支持
训练策略

项目资源

物料清单 (BOM)

物料名称 数量 参考价格 备注
Mobile ALOHA 机器人 1 真实部署需要
工作站/服务器 1 训练和推理用,建议配备GPU
Ubuntu/Linux系统 1 推荐使用
Conda 1 环境管理
PyTorch 1 深度学习框架
MuJoCo 2.3.7 1 物理模拟引擎
DM_Control 1.0.14 1 控制库
robomimic (r2d2分支) 1 Diffusion Policy依赖

3D 模型

所需工具

工具用途是否必需
GPU (NVIDIA) 训练和推理深度学习模型 ✅ 是
摄像头 真实机器人数据采集 ▢ 推荐
数据采集设备 收集人类演示数据 ▢ 推荐

能力画像

记忆与知识检索
逻辑推演
表达与交流
感知与观察
数理与计算
动手与操作
狂热与坚持
创造与创新

所需技能

🔧 **动手能力**:需要组装和调试 Mobile ALOHA 机器人硬件系统,进行真实环境的数据采集和部署 💻 **编程能力**:需要 Python 编程经验,理解深度学习框架(PyTorch),熟悉命令行操作和环境配置 ⚡ **电子电路**:需要基本的机器人硬件知识,了解传感器和执行器的连接与通信

适用场景

机器人模仿学习算法研究与开发
双臂协作任务的自动化(如转移物体、插入操作)
基于人类演示的机器人技能学习
移动操作机器人(Mobile Manipulator)的策略训练
学术研究:比较不同模仿学习算法在机器人任务上的表现