wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Cleanup

This commit is contained in:
Simon Alibert 2024-10-17 16:08:37 +02:00
parent 3ee3739edc
commit 7242c57400
1 changed files with 0 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
lerobot/common/datasets/v2/batch_convert_dataset_v1_to_v2.py
View File

@ -100,8 +100,6 @@ ALOHA_SINGLE_TASKS_REAL = {
"aloha_mobile_cabinet": "Open the top cabinet, store the pot inside it then close the cabinet.", "aloha_mobile_cabinet": "Open the top cabinet, store the pot inside it then close the cabinet.",
"aloha_mobile_chair": "Push the chairs in front of the desk to place them against it.", "aloha_mobile_chair": "Push the chairs in front of the desk to place them against it.",
"aloha_mobile_elevator": "Take the elevator to the 1st floor.", "aloha_mobile_elevator": "Take the elevator to the 1st floor.",
# Alternative version, not sure what's best.
# 'aloha_mobile_elevator': "Navigate to the elevator and call it. When it arrives, get inside et push the 1st floor button.",
"aloha_mobile_shrimp": "Sauté the raw shrimp on both sides, then serve it in the bowl.", "aloha_mobile_shrimp": "Sauté the raw shrimp on both sides, then serve it in the bowl.",
"aloha_mobile_wash_pan": "Pick up the pan, rinse it in the sink and then place it in the drying rack.", "aloha_mobile_wash_pan": "Pick up the pan, rinse it in the sink and then place it in the drying rack.",
"aloha_mobile_wipe_wine": "Pick up the wet cloth on the faucet and use it to clean the spilled wine on the table and underneath the glass.", "aloha_mobile_wipe_wine": "Pick up the wet cloth on the faucet and use it to clean the spilled wine on the table and underneath the glass.",
@ -133,13 +131,6 @@ def batch_convert():
name = repo_id.split("/")[1] name = repo_id.split("/")[1]
single_task, tasks_col, robot_config = None, None, None single_task, tasks_col, robot_config = None, None, None
# TODO(aliberts) issues with these datasets:
# if name in [
# "aloha_mobile_shrimp", # 18 videos files per camera but 17 episodes in the parquet
# # "aloha_mobile_wash_pan", #
# ]:
# continue
if "aloha" in name: if "aloha" in name:
robot_config = parse_robot_config(ALOHA_CONFIG) robot_config = parse_robot_config(ALOHA_CONFIG)
if "sim_insertion" in name: if "sim_insertion" in name: