From a091b0d3443abb38799fd64d2ad731d3dfda2ae8 Mon Sep 17 00:00:00 2001 From: Mishig Davaadorj Date: Tue, 21 May 2024 15:03:03 +0200 Subject: [PATCH] [visualize_dataset.py] Load datasets from hf.co rather than local --- lerobot/scripts/visualize_dataset.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lerobot/scripts/visualize_dataset.py b/lerobot/scripts/visualize_dataset.py index 8af909fd..930e88b2 100644 --- a/lerobot/scripts/visualize_dataset.py +++ b/lerobot/scripts/visualize_dataset.py @@ -236,8 +236,8 @@ def write_episode_data_html(output_dir, file_name, js_fname, ep_index, dataset): first_frame_idx = dataset.episode_data_index["from"][ep_index].item() for key in dataset.video_frame_keys: - # Example of video_path: 'videos/observation.image_episode_000004.mp4' - video_path = dataset.hf_dataset.select_columns(key)[first_frame_idx][key]["path"] + # Example of video_path: 'https://huggingface.co/datasets/{dataset.repo_id}/resolve/{dataset.version}/videos/observation.image_episode_000004.mp4' + video_path = f"https://huggingface.co/datasets/{dataset.repo_id}/resolve/{dataset.version}/" + dataset.hf_dataset.select_columns(key)[first_frame_idx][key]["path"] videos_div.append(create_video(f"video_{key}", video_path)) # Add controls for videos and graph