⬆️ upgrade lerobot to 0.2.0 (#51)

This commit is contained in:
Qizhi Chen
2025-07-31 08:51:51 +08:00
committed by GitHub
parent 8990cafd40
commit 858cbd582d
13 changed files with 27 additions and 27 deletions
+2 -2
View File
@@ -1090,7 +1090,7 @@ def early_validation(source_folders, episode_mapping, default_fps=20, fps=None):
print(f"{colored('WARNING', 'yellow', attrs=['bold'])}: Video file not found for {image_dir}, episode {old_index} in {old_folder}")
if image_dir_exists:
print(" Image directory exists, encoding video from images.")
from lerobot.common.datasets.video_utils import encode_video_frames
from lerobot.datasets.video_utils import encode_video_frames
encode_video_frames(source_image_dir, source_video_path, fps, overwrite=True)
print(" Encoded video frames successfully.")
else:
@@ -1124,7 +1124,7 @@ def early_validation(source_folders, episode_mapping, default_fps=20, fps=None):
print(f" Expected: {expected_frames}, Found: {video_frame_count}")
print(f" Re-encoded video frames from {source_image_dir} to {source_video_path}")
from lerobot.common.datasets.video_utils import encode_video_frames
from lerobot.datasets.video_utils import encode_video_frames
encode_video_frames(source_image_dir, source_video_path, fps, overwrite=True)
print(" Re-encoded video frames successfully.")