
LeRobot 机器人控制框架完整教程20 分钟从环境到训练出第一个策略模型【免费下载链接】lerobot LeRobot: Making AI for Robotics more accessible with end-to-end learning项目地址: https://gitcode.com/GitHub_Trending/le/lerobot想让机器人学会抓取你通常要自己写训练循环、对接各家硬件驱动、把散落的数据拼成模型能吃的格式。LeRobot 把这三件事都做好了它是 Hugging Face 出品的 Python 原生机器人 AI 控制框架统一了机器人控制接口、数据集格式和主流策略模型。跟着本文走完 4 条命令、约 20 分钟你就能完成环境检查、读懂一个真实机器人数据集、再训练出第一个策略模型——全程不用碰底层驱动代码。 第一步LeRobot 一键安装步骤与验证这一节能帮你把开发环境一次装对装好约 3 分钟。LeRobot 要求 Python 3.12 及以上版本先确认自己的 Python 版本达标再依次执行pip install lerobot lerobot-info第一条命令安装核心包并自动带上 PyTorch 等依赖第二条会打印版本、Python 版本和可用的加速设备。如果你能在终端看到清晰的版本信息而没有报错环境就准备就绪了后面每一步都依赖它。 最快读懂第一个机器人数据集这一节能让你 30 秒内上手一个真实的机器人数据集为训练做准备。LeRobotDataset 把 MP4 视频视觉和 Parquet 表状态、动作统一存放并直接托管在 Hugging Face Hub 上意味着你不用自己采数据就能开练。打开任意一个 Python 脚本运行from lerobot.datasets.lerobot_dataset import LeRobotDataset dataset LeRobotDataset(lerobot/aloha_mobile_cabinet) print(dataset[0][action].shape)它会从 Hub 拉取 ALOHA 机械臂数据集并自动处理好视频解码你会看到终端打印出动作向量的形状比如(7,)。数据读取完成后就可以直接进入下一步训练。关于 v3 格式的目录布局、流式加载等细节见 LeRobotDataset 文档。 一条命令训练你的第一个机器人策略模型这一节能帮你用 1 行命令启动完整训练你只需要配置参数。LeRobot 内置了 ACT、Diffusion、VQ-BeT、Pi0、GR00T 等策略选一个即可lerobot-train \ --policy.typeact \ --dataset.repo_idlerobot/aloha_mobile_cabinet它会自动下载数据、做归一化预处理、搭建模型并开始训练终端会滚动打印每一步的 loss 变化。下表是新手最常改的 3 个参数按需替换即可参数含义新手推荐--policy.type策略模型类型act训练快、效果稳--dataset.repo_id数据集仓库 ID用现成的公开数据集--batch_size每批样本数显存不足时调小训练跑完你会得到一个可以直接用于评估的策略模型目录。更多模型对比和各策略的显存要求可看 策略源码 与 计算硬件指南。 动手采数据遥操作录制自己的数据集这一节能让你从用别人的数据升级为造自己的数据这是让策略真正学会你任务的关键。把 SO101 机械臂和摄像头接上用下面这条命令开始遥操作录制lerobot-record \ --robot.typeso101_follower \ --teleop.typeso101_leader \ --dataset.repo_id${HF_USER}/my-dataset \ --dataset.num_episodes5 \ --dataset.single_taskgrab the cube它会自动连接跟随臂和领导臂、启动摄像头你只需动领导臂动作就会实时编码成 MP4 和 Parquet 存下来录满 5 条 episode 后数据直接推送到你的仓库拿来就能训练。摄像头连不上时先跑lerobot-find-cameras确认设备编号。完整的机器人硬件接入指南在 硬件集成文档常用命令速查见 使用文档。✅ 到此为止你手里已经有什么回看这 4 步环境已装好并通过lerobot-info验证、你能加载并读懂标准数据集、能用一条命令训出策略模型、还能自己遥操作采集新数据——这正是 LeRobot 的完整闭环采集、训练、评估一条龙。下一步就去跑一次lerobot-record用你的机械臂录下第一个真实任务你会对机器人 AI从概念变成手感。【免费下载链接】lerobot LeRobot: Making AI for Robotics more accessible with end-to-end learning项目地址: https://gitcode.com/GitHub_Trending/le/lerobot创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考