admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-24 18:26:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

add torchcodec cpu

Browse Source
This commit is contained in:
root
2025-03-02 20:47:33 +00:00
parent 8861546ad8
commit 4e2dc91e59
2 changed files with 64 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
lerobot/common/datasets/lerobot_dataset.py
+3 -3
View File
@@ -67,7 +67,7 @@ from lerobot.common.datasets.utils import (
)
from lerobot.common.datasets.video_utils import (
VideoFrame,
decode_video_frames_torchvision,
decode_video_frames_torchcodec,
encode_video_frames,
get_video_info,
)
@@ -707,8 +707,8 @@ class LeRobotDataset(torch.utils.data.Dataset):
item = {}
for vid_key, query_ts in query_timestamps.items():
video_path = self.root / self.meta.get_video_file_path(ep_idx, vid_key)
frames = decode_video_frames_torchvision(
video_path, query_ts, self.tolerance_s, self.video_backend
frames = decode_video_frames_torchcodec(
video_path, query_ts, self.tolerance_s
)
item[vid_key] = frames.squeeze(0)