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

fix windows html visualize to make video could be seen

This commit is contained in:
HuangJunshan 2025-01-17 23:44:55 +08:00
parent 380b836eee
commit ff60cd1ca9
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lerobot/scripts/visualize_dataset_html.py
View File

@ -174,7 +174,7 @@ 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"]
@ -369,7 +369,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)