mirror of
https://github.com/Tavish9/any4lerobot.git
synced 2026-05-27 19:49:41 +00:00
⬆️ upgrade lerobot to 0.2.0 (#51)
This commit is contained in:
@@ -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.")
|
||||
|
||||
|
||||
Reference in New Issue
Block a user