LeRobot机器人入门教程:SO-100机械臂从装好到练出抓取策略的完整上手指南
【免费下载链接】lerobot🤗 LeRobot: Making AI for Robotics more accessible with end-to-end learning项目地址: https://gitcode.com/GitHub_Trending/le/lerobot
想让机械臂抓个杯子,你可能要先跟舵机串口协议、相机取帧同步、数据集格式对齐这些底层细节搏斗一晚上——机器人学习的门槛从来不在算法,而在工程链路。LeRobot 是 Hugging Face 开源的 PyTorch 机器人学习库,把硬件控制、数据集、策略训练、仿真评测整条链路都打通了,你只需要pip install lerobot就能开始。本文带你用一套 SO-100 低成本机械臂(几百块钱的 3D 打印关节臂)走通"采数据 → 训策略 → 做评测"的最小闭环 🚀
零硬件门槛:装好环境并验证LeRobot可用
不用先买硬件。LeRobot 直接发布在 PyPI,安装分两个动作:
pip install "lerobot[core_scripts]" lerobot-infolerobot-info会打印你的环境状态,跑通它说明安装没问题(要求 Python ≥ 3.12、PyTorch ≥ 2.10,官方推荐用 conda 建独立环境,详见 docs/source/installation.mdx)。后面的章节会用到lerobot-find-port(找串口)、lerobot-record(录数据集)、lerobot-train(训练)这些 CLI 命令,它们都随core_scripts一起装好。没有 GPU 也不怕:先跑仿真评测和看社区预训练模型,真机训练再上卡。
把SO-100接进来:找端口和电机配置
SO-100 由 leader(你手扶的主臂)和 follower(被学习的从臂)两条 SO-100 组成,USB 接上电脑后第一件事是找到各自的串口:
- 运行
lerobot-find-port,按提示拔掉/插回某条臂的 USB 线,程序会锁定对应的端口路径 - 电机出厂默认 ID 都是 1,必须在装臂之前先给每个电机烧唯一 ID 并统一波特率,SO-100 装好后舵机接线口就不好拆了 ⚠️
完整装配和配置步骤在 docs/source/so100.mdx,Feetech 舵机驱动需要额外装 extras:pip install -e ".[feetech]"(源码树里对应 src/lerobot/motors/feetech/)。
遥操作采集第一个LeRobot数据集
有了两条臂,数据就靠"人做、机器学"来采:你扶 leader 臂做"抓方块"动作,follower 臂跟着动,LeRobot 把每一步的关节状态、动作和相机画面同步录下来。核心就是一条lerobot-record命令,大致这样:
lerobot-record \ --robot.type=so100_follower --robot.port=/dev/ttyACM0 \ --teleop.type=so100_leader --teleop.port=/dev/ttyACM1 \ --dataset.repo_id=<你的用户名>/<数据集名> \ --dataset.single_task="Grab the cube"采集过程支持实时可视化回显,录完的数据采用LeRobotDataset格式(Parquet 存状态/动作 + MP4 存视频,见 docs/source/lerobot-dataset-v3.mdx),可以直接推到 Hugging Face Hub。关键点:Hub 上有数千个社区数据集和训练好的模型,你完全可以先不录数据,直接拿现成的练手 🎥
训练ACT策略并用仿真环境做评测
训练一条策略只需要两行参数,以 ACT(Action Chunking with Transformers,模仿学习里把动作切成连续小块来预测的策略)为例:
lerobot-train \ --policy.type=act \ --dataset.repo_id=lerobot/aloha_mobile_cabinet模型库里不止 ACT:模仿学习有 Diffusion、VQ-BeT,视觉语言动作(VLA)模型有 Pi0、GR00T、SmolVLA 等十几个,全部纯 PyTorch 实现,清单和每个策略的说明在 docs/source/ 下。训完(或直接用 Hub 上的现成模型)后,用lerobot-eval跑仿真基准验证效果:指定策略路径、环境类型(如 LIBERO、MetaWorld)、回合数,它会自动跑 rollout 并算出成功率,比一上来就怼真机省钱得多 📊
常见坑与进阶:权限、时序和第三方硬件
三个新手最常踩的坑:
- Linux 串口权限:
/dev/ttyACM0这类端口默认没有读写权限,先执行sudo chmod 666 /dev/ttyACM0,否则报 permission denied(见 src/lerobot/robots/so_follower/so100.md) - WSL 用户:在 Windows 下用 WSL 跑时记得补装
evdev依赖,键盘类遥操作会用到 - 电机配置时序:重复一遍——SO-100 的电机 ID/波特率必须在总装前配好,配反了得拆臂重来
进阶方向两条:一是第三方硬件,LeRobot 用包名约定做插件自动发现,任何lerobot_robot_<name>前缀的 pip 包装上就能直接用在 CLI 里,社区已有 xArm、UR5e、Franka、WidowX 等插件;二是 src/lerobot/policies/ 里每个策略目录都带 README,照着 bring your own policy 的思路可以把自己的模型接进同一套采集、训练、评测工具链。
【免费下载链接】lerobot🤗 LeRobot: Making AI for Robotics more accessible with end-to-end learning项目地址: https://gitcode.com/GitHub_Trending/le/lerobot
创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考