Compare commits

...

5 Commits

Author SHA1 Message Date
Junshan Huang b7f35ab92d
Merge 06a8e55f29 into 78fd1a1e04 2025-03-28 13:15:29 +08:00
pre-commit-ci[bot] 06a8e55f29 [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
2025-03-21 08:39:03 +00:00
Simon Alibert 0859190f82
Merge branch 'main' into main 2025-03-21 09:38:48 +01:00
Junshan Huang dacb57ad8a
Merge branch 'main' into main 2025-02-06 18:54:12 +08:00
HuangJunshan ff60cd1ca9 fix windows html visualize to make video could be seen 2025-01-17 23:46:00 +08:00
1 changed files with 5 additions and 2 deletions

View File

@ -174,7 +174,10 @@ def run_server(
dataset.meta.get_video_file_path(episode_id, key) for key in dataset.meta.video_keys dataset.meta.get_video_file_path(episode_id, key) for key in dataset.meta.video_keys
] ]
videos_info = [ videos_info = [
{"url": url_for("static", filename=video_path), "filename": video_path.parent.name} {
"url": url_for("static", filename=str(video_path).replace("\\", "/")),
"filename": video_path.parent.name,
}
for video_path in video_paths for video_path in video_paths
] ]
tasks = dataset.meta.episodes[episode_id]["tasks"] tasks = dataset.meta.episodes[episode_id]["tasks"]
@ -381,7 +384,7 @@ def visualize_dataset_html(
if isinstance(dataset, LeRobotDataset): if isinstance(dataset, LeRobotDataset):
ln_videos_dir = static_dir / "videos" ln_videos_dir = static_dir / "videos"
if not ln_videos_dir.exists(): if not ln_videos_dir.exists():
ln_videos_dir.symlink_to((dataset.root / "videos").resolve()) ln_videos_dir.symlink_to((dataset.root / "videos").resolve().as_posix())
if serve: if serve:
run_server(dataset, episodes, host, port, static_dir, template_dir) run_server(dataset, episodes, host, port, static_dir, template_dir)