admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-20 02:59:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix bug: correctly specify paths

Browse Source
This commit is contained in:
fracapuano
2025-11-07 20:59:58 +00:00
parent c1f5d8f48f
commit 1612aa7ac7
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/behavior_1k/convert_to_lerobot_v3.py
+2 -2
View File
@@ -174,7 +174,7 @@ def get_image_keys(root):
def convert_data(root: Path, new_root: Path, data_file_size_in_mb: int, task_index: int): def convert_data(root: Path, new_root: Path, data_file_size_in_mb: int, task_index: int):
task_dir_name = f"task-00{task_index}" task_dir_name = f"task-{task_index:04d}"
data_dir = root / "data" / task_dir_name data_dir = root / "data" / task_dir_name
ep_paths = sorted(data_dir.glob("*.parquet")) ep_paths = sorted(data_dir.glob("*.parquet"))
image_keys = get_image_keys(root) image_keys = get_image_keys(root)
@@ -230,7 +230,7 @@ def convert_videos_of_camera(
# Access old paths to mp4 # Access old paths to mp4
# videos_dir = root / "videos" # videos_dir = root / "videos"
# ep_paths = sorted(videos_dir.glob(f"*/{video_key}/*.mp4")) # ep_paths = sorted(videos_dir.glob(f"*/{video_key}/*.mp4"))
task_dir_name = f"task-00{task_index}" task_dir_name = f"task-{task_index:04d}"
videos_dir = root / "videos" / task_dir_name / video_key videos_dir = root / "videos" / task_dir_name / video_key
ep_paths = sorted(videos_dir.glob("*.mp4")) ep_paths = sorted(videos_dir.glob("*.mp4"))
print("ep_paths", ep_paths) print("ep_paths", ep_paths)