5.2 Host/Client 模式记录
5.2 Host/Client 模式记录
无线版 LeKiWi 使用 Host/Client 架构,树莓派运行 lekiwi_host,桌面端运行 record.py 进行远程数据采集。
架构说明
[树莓派] lekiwi_host ←→ tcp://<IP>:5555 ←→ [桌面端] record.py
修改 record.py — 三处代码修改
在桌面端打开 examples/lekiwi/record.py,修改以下三处配置:
第 1 处:LeKiWiClientConfig — 改为树莓派 IP
# 修改前(约第 37 行)
robot_config = LeKiWiClientConfig(remote_ip="192.168.31.165", id="my_lekiwi")
# 修改后(替换为树莓派实际 IP)
robot_config = LeKiWiClientConfig(remote_ip="192.168.x.x", id="my_lekiwi")
第 2 处:SO101LeaderConfig — 确认本机串口
Leader 臂连接在桌面端,确认 port 为桌面端的实际串口(运行 lerobot-find-port 查找):
# 修改前(约第 41 行,示例端口)
leader_arm_config = SO101LeaderConfig(port="COM69", id="R07252710")
第 3 处:id — 改为你自己的编号
将两处 id 改为你自己的标定编号(在舵机标定章节中获取)。
确认录制参数
HF_REPO_ID、TASK_DESCRIPTION 和 NUM_EPISODES 按实际需求配置。
树莓派端启动 host
SSH 登录树莓派,启动 lekiwi_host(以编号 R12252801 为例,请替换为自己的 Follower 臂编号):
ssh joyandai@<树莓派IP>
conda activate lerobot
python -m lerobot.robots.lekiwi.lekiwi_host --robot.id R12252801
桌面端启动录制
conda activate lerobot
python examples/lekiwi/record.py
录制时会打开 Rerun 窗口实时显示摄像头画面和电机状态。
数据采集建议
- 起始任务:抓取一个物体放到不同位置的盒子中
- 建议数量:至少 50 个 episode,每个位置 10 个
- 保持一致性:相机位置固定,抓取动作一致
- 可视化验证:确保只看相机画面你也能完成任务
- 渐进复杂度:先跑通基础抓取,再逐步增加位置和姿态变化
一个好的经验法则:先录制 10 组数据跑通全流程,确认训练有效果后再扩充数据集。
树莓派端启动 host
linux: ssh joyandai@192.168.x.x
conda activate lerobot
python -m lerobot.robots.lekiwi.lekiwi_host --robot.id R12252801
macos: ssh joyandai@192.168.x.x
conda activate lerobot
python -m lerobot.robots.lekiwi.lekiwi_host --robot.id R12252801
windows: ssh joyandai@192.168.x.x
conda activate lerobot
python -m lerobot.robots.lekiwi.lekiwi_host --robot.id R12252801
桌面端远程记录
linux: python examples/lekiwi/record.py
macos: python examples/lekiwi/record.py
windows: python examples\lekiwi\record.py