mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-18 10:10:08 +00:00
@@ -286,7 +286,7 @@ def convert_videos(root: Path, new_root: Path, video_file_size_in_mb: int):
|
|||||||
if len(set(num_eps_per_cam)) != 1:
|
if len(set(num_eps_per_cam)) != 1:
|
||||||
raise ValueError(f"All cams dont have same number of episodes ({num_eps_per_cam}).")
|
raise ValueError(f"All cams dont have same number of episodes ({num_eps_per_cam}).")
|
||||||
|
|
||||||
episods_metadata = []
|
episodes_metadata = []
|
||||||
num_cameras = len(video_keys)
|
num_cameras = len(video_keys)
|
||||||
num_episodes = num_eps_per_cam[0]
|
num_episodes = num_eps_per_cam[0]
|
||||||
for ep_idx in tqdm.tqdm(range(num_episodes), desc="convert videos"):
|
for ep_idx in tqdm.tqdm(range(num_episodes), desc="convert videos"):
|
||||||
@@ -299,9 +299,9 @@ def convert_videos(root: Path, new_root: Path, video_file_size_in_mb: int):
|
|||||||
ep_dict = {}
|
ep_dict = {}
|
||||||
for cam_idx in range(num_cameras):
|
for cam_idx in range(num_cameras):
|
||||||
ep_dict.update(eps_metadata_per_cam[cam_idx][ep_idx])
|
ep_dict.update(eps_metadata_per_cam[cam_idx][ep_idx])
|
||||||
episods_metadata.append(ep_dict)
|
episodes_metadata.append(ep_dict)
|
||||||
|
|
||||||
return episods_metadata
|
return episodes_metadata
|
||||||
|
|
||||||
|
|
||||||
def convert_videos_of_camera(root: Path, new_root: Path, video_key: str, video_file_size_in_mb: int):
|
def convert_videos_of_camera(root: Path, new_root: Path, video_key: str, video_file_size_in_mb: int):
|
||||||
|
|||||||
Reference in New Issue
Block a user