hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

Use pusht as example so it's fast

Browse Source
This commit is contained in:
Cadene
2024-04-11 13:36:50 +00:00
parent 2f4af32d3f
commit a605eec7e9
2 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/1_visualize_dataset.py
View File

@@ -9,7 +9,7 @@ print(lerobot.available_datasets)
# >>> ['aloha_sim_insertion_human', 'aloha_sim_insertion_scripted', 'aloha_sim_transfer_cube_human', 'aloha_sim_transfer_cube_scripted', 'pusht', 'xarm_lift_medium']
# TODO(rcadene): remove DATA_DIR
dataset = AlohaDataset("aloha_sim_transfer_cube_human", root=Path(os.environ.get("DATA_DIR")))
dataset = AlohaDataset("pusht", root=Path(os.environ.get("DATA_DIR")))
video_paths = render_dataset(
dataset,
@@ -17,4 +17,4 @@ video_paths = render_dataset(
max_num_episodes=1,
)
print(video_paths)
# ['outputs/visualize_dataset/example/episode_0_top.mp4']
# ['outputs/visualize_dataset/example/episode_0.mp4']