Skip to content

Update convert_aloha_data_to_lerobot.py#723

Open
MSTLE wants to merge 1 commit intoPhysical-Intelligence:mainfrom
MSTLE:patch-1
Open

Update convert_aloha_data_to_lerobot.py#723
MSTLE wants to merge 1 commit intoPhysical-Intelligence:mainfrom
MSTLE:patch-1

Conversation

@MSTLE
Copy link

@MSTLE MSTLE commented Oct 14, 2025

适配 lerobot新版本3.0+
脚本修改总结:
将导入从 lerobot.common.datasets.lerobot_dataset.LEROBOT_HOME 改为 lerobot.common.constants.HF_LEROBOT_HOME 移除了不存在的 download_raw 导入
删除了 port_aloha() 函数的 raw_repo_id 参数
将所有 LEROBOT_HOME 引用替换为 HF_LEROBOT_HOME(共 4 处)
在 create_empty_dataset() 中添加 cameras 参数,移除硬编码的相机列表 在 load_raw_episode_data() 中添加 cameras 参数,实现动态相机支持
在 populate_dataset() 中添加 cameras 参数并传递给子函数
在 port_aloha() 中添加 cameras = get_cameras(hdf5_files) 自动检测相机 在每个 frame 字典中添加 "task": task 字段
将 dataset.save_episode(task=task) 改为 dataset.save_episode() 移除了 dataset.consolidate() 调用
核心变化:适配 LeRobot 新版本 API,支持自动检测相机配置

适配 lerobot新版本3.0+
脚本修改总结:
将导入从 lerobot.common.datasets.lerobot_dataset.LEROBOT_HOME 改为 lerobot.common.constants.HF_LEROBOT_HOME
移除了不存在的 download_raw 导入
删除了 port_aloha() 函数的 raw_repo_id 参数
将所有 LEROBOT_HOME 引用替换为 HF_LEROBOT_HOME(共 4 处)
在 create_empty_dataset() 中添加 cameras 参数,移除硬编码的相机列表
在 load_raw_episode_data() 中添加 cameras 参数,实现动态相机支持
在 populate_dataset() 中添加 cameras 参数并传递给子函数
在 port_aloha() 中添加 cameras = get_cameras(hdf5_files) 自动检测相机
在每个 frame 字典中添加 "task": task 字段
将 dataset.save_episode(task=task) 改为 dataset.save_episode()
移除了 dataset.consolidate() 调用
核心变化:适配 LeRobot 新版本 API,支持自动检测相机配置
@jimmyt857 jimmyt857 removed their request for review October 14, 2025 15:46
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant