LeRobot数据采集全流程解析：从环境配置到动作回放（SO-100实战）
LeRobot数据采集全流程实战：从环境搭建到动作复现的SO-100深度指南
当我们需要让机器人学会新技能时，数据采集是构建智能系统的第一步。LeRobot作为Hugging Face推出的机器人学习平台，通过标准化流程降低了开发门槛。本文将带你完整走通SO-100型号机器人的数据采集闭环，涵盖从环境准备到动作回放的全套实践方案。
1. 开发环境配置与初始化
在开始采集数据前，需要搭建支持LeRobot运行的开发环境。推荐使用Ubuntu 22.04 LTS系统，这是经过社区验证最稳定的基础环境。
1.1 基础依赖安装
首先通过conda创建隔离的Python环境：
conda create -n lerobot python=3.10 -y conda activate lerobot接着克隆项目仓库并安装核心依赖：
git clone https://github.com/huggingface/lerobot.git cd lerobot && pip install -e ".[feetech]"注意：必须安装指定版本的OpenCV（≥4.10.0），否则会导致视频采集异常：
conda install -y -c conda-forge "opencv>=4.10.0" ffmpeg1.2 硬件驱动配置
SO-100机器人需要额外安装Feetech串口驱动：
sudo apt-get install libserial-dev # Ubuntu系统 pip install pyserial==3.5连接机器人后，可通过以下命令测试通信：
python -c "from serial.tools import list_ports; print(list_ports.comports())"2. 数据采集全流程详解
2.1 初始化采集参数
数据采集的核心命令是control_robot.py record，关键参数包括：
| 参数 | 类型 | 说明 | 默认值 |
|---|---|---|---|
| --fps | int | 采集帧率 | 30 |
| --episode-time-s | int | 单次采集时长(秒) | 40 |
| --num-episodes | int | 采集次数 | 2 |
| --warmup-time-s | int | 设备预热时间 | 5 |
典型启动命令示例：
python lerobot/scripts/control_robot.py record \ --robot-path lerobot/configs/robot/so100.yaml \ --fps 30 \ --repo-id your_hf_username/so100_dataset \ --tags "demo" \ --episode-time-s 602.2 增量采集模式
当需要追加数据时，使用--resume参数控制采集行为：
# 覆盖已有数据集（危险！） --force-override 1 # 增量模式（推荐） --resume 1 --local-files-only 1重要区别：
force-override会清空现有数据resume会在现有数据上追加新记录
2.3 数据质量控制
采集过程中建议实时监控数据质量：
- 检查时间戳连续性：
grep "timestamp" data/raw/*.json - 验证图像完整性：
ffmpeg -i data/videos/episode_0.mp4 -vf select='eq(pict_type,I)' -vsync vfr keyframes-%03d.png - 检测动作范围：
python scripts/check_joint_limits.py --data-dir data
3. 数据集管理与版本控制
3.1 本地数据集结构
采集完成后会生成标准目录结构：
data/ ├── meta.json # 元数据 ├── calibration/ # 标定文件 ├── videos/ # 视频流 └── trajectories/ # 关节轨迹3.2 上传至Hugging Face Hub
使用以下命令将数据集发布到社区：
huggingface-cli upload your_hf_username/so100_dataset ./data --token YOUR_TOKEN上传前建议添加数据集卡片：
--- tags: [robotics, demonstration] license: apache-2.0 --- # SO-100 Pick-and-Place Dataset 包含20组抓放动作的示教数据，每组包含： - 1280x720 RGB视频 - 6轴关节角度记录 - 末端执行器状态4. 动作回放与验证
4.1 网页可视化
启动本地数据查看服务：
python lerobot/scripts/visualize_dataset_html.py \ --repo-id your_hf_username/so100_dataset \ --port 9090访问http://localhost:9090可交互式查看：
- 视频流与关节数据同步播放
- 三维轨迹可视化
- 数据统计图表
4.2 物理回放控制
在真实机器人上复现动作：
python lerobot/scripts/control_robot.py replay \ --robot-path lerobot/configs/robot/so100.yaml \ --episode 0 \ --speed 0.8 # 降速模式更安全关键安全措施：
- 首次运行时添加
--dry-run参数进行虚拟测试 - 工作空间清空无关物体
- 急停开关保持可触达状态
5. 典型问题排查指南
5.1 采集中断恢复
当采集过程意外终止时：
- 检查最后有效帧时间戳：
tail -n 50 data/logs/record.log | grep "Saved frame" - 使用增量模式继续采集：
--resume 1 --start-frame 420
5.2 数据同步异常
若发现传感器数据不同步：
- 重新校准时间戳：
from lerobot.utils import time_align time_align.calibrate_offsets(data_dir) - 检查系统时钟源：
chronyc sources # 需要NTP服务正常
5.3 网络连接问题
遇到Hugging Face连接超时：
- 测试API连通性：
curl -I https://huggingface.co - 临时使用本地缓存模式：
--local-files-only 1
在实际项目中，我们发现SO-100的关节数据采集偶尔会出现±2°的抖动，这通常是由于电源干扰导致。建议为控制柜单独配置稳压电源，并在数据分析阶段加入低通滤波处理。