IT之家 2024年12月17日
宇树机器人强化学习代码全面开源,还有训练到仿真和实操手把手教学
index_new5.html
../../../zaker_core/zaker_tpl_static/wap/tpl_guoji1.html

 

宇树机器人开源了其强化学习代码,不仅支持仿真运行,还能部署到实体机器人。此次更新增加了对MuJoCo模拟仿真的支持,并可迁移至H1、H1-2和G1三款机器人。该项目名为RL GYM,提供了从训练到模拟再到真机运行的完整开源体系,并附有详细教程。此外,宇树还开源了ROS模拟包、Vision Pro遥操作控制代码,以及人形机器人操作数据集和激光雷达算法。这一系列开源举措,无疑加速了机器人技术的发展。

🚀宇树机器人开源RL GYM项目,实现从训练到真机部署的完整流程,支持Issac Gym和MuJoCo仿真环境,并提供详细教程。

🤖RL GYM支持宇树H1、H1-2和G1三款人形机器人,及机器狗Go2的训练和模拟,用户可基于此进行二次开发和研究。

🧰宇树开源了ROS模拟包、Vision Pro遥操作控制代码以及人形机器人操作数据集,促进了机器人技术的广泛应用。

💡宇树还开源了Point-LIO激光雷达算法,进一步丰富了其开源体系,为机器人感知技术的发展贡献力量。

9.9 万元起,还能够大规模量产的国产人形机器人,表现得实在是太 6 了:

而且还走上了开源路线,就在刚刚,宇树机器人开源的强化学习代码又更新了!

更新之后不再是只能训练 —— 不仅能够仿真运行,还能部署到实体机器,整个过程所有代码全部开放

一开始,宇树开源的是英伟达 Issac Gym 平台上的训练代码,这次新增了对 MuJoCo 模拟仿真的支持。

而且还能迁移到实体机器人上运行,目前支持宇树的 H1、H1-2 和 G1 三款机器人。

这下从训练到模拟再到真机运行,整个 RL 套件的开源体系终于完整了。

看到这个消息,有网友激动地表示,训练机器人灵魂的方法终于公开了!

全过程代码公开,还有详细教程

宇树给这个项目取名叫 RL GYM,可能和一开始专门提供基于 Issac Gym 的训练代码有关。

现在 RL GYM 又支持了 MuJoCo,可以在预训练的基础上进行仿真模拟了。

训练阶段的 Issac Gym 需要 CUDA,也就是需要 N 卡,仿真阶段的 MuJoCo 则支持各种 GPU,甚至 CPU 和 TPU 也能运行。

从环境的安装配置,到训练和模拟,以及最后的真机迁移,不仅有代码,还有非常详细的手把手教程。

以最新款的 G1 为例,在 Issac Gym 里训练完之后的效果是这样的:

放到 MuJoCo 里模拟,就有了开头看到在这段 DEMO:

最后迁移到 G1 真机,就能看到训练的效果了(并且真机迁移的部分还有中文教程):

H1 和 H1-2 也与此类似,另外通过 RL GYM 还可以在 Issac Gym 里训练机器狗 Go2:

还有更多开源项目

在宇树的整个开源体系当中,RL GYM 只是其中一环,在宇树所有的 GitHub 仓库中星标数排第三。

排第一的是针对开源机器人系统 R (obot) OS 推出的模拟包,包含了宇树系列机器人的质量、惯量、力矩等参数。

星标数第二的则是使用苹果 Vision Pro 对 G1 进行遥操作控制的仓库,可以用于数据采集。

采集到的数据是 JSON 形式,而训练通过 LeRobot 实现,因此宇树还提供了将 JSON 格式数据转为 LeRobot 所需格式的教程。

其中包含遥操作控制的代码教程,以及硬件配置图、物料清单和安装说明:

除了 GitHub 上发布的这些工具,宇树还在 HuggingFace 上公开了训练数据集,与数据采集工具同期发布。

数据集包含五种操作,使用配备有三指灵巧手的宇树 G1 人形机器人收集,每张图分辨率为 640x480,每个手臂和灵巧手的状态和动作维度为 7。

比如,利用这套数据集可以让机器人学习拧瓶盖倒水:

将三色积木叠放到一起:

此外还包括将摄像头放入相应包装盒、收集物品并存储、双臂抓取红色木块并将其放入一个黑色长方形容器中这些操作。

最后,宇树开源的也不只是和机器人相关的项目。

上个月宇树发布了售价 419 美元的激光雷达,在产品上线的同时就将其采用的 Point-LIO 算法进行了开源。

项目仓库:

https://github.com/unitreerobotics/unitree_rl_gym

Fish AI Reader

Fish AI Reader

AI辅助创作,多种专业模板,深度分析,高质量内容生成。从观点提取到深度思考,FishAI为您提供全方位的创作支持。新版本引入自定义参数,让您的创作更加个性化和精准。

FishAI

FishAI

鱼阅,AI 时代的下一个智能信息助手,助你摆脱信息焦虑

联系邮箱 441953276@qq.com

相关标签

人形机器人 开源 强化学习 宇树机器人 ROS
相关文章