wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update lerobot/common/datasets/video_utils.py 

Co-authored-by: Remi <re.cadene@gmail.com>
This commit is contained in:
Jade Choghari 2025-03-04 13:27:24 +03:00 committed by GitHub
parent a8fcd3512d
commit c03b0db8aa
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lerobot/common/datasets/video_utils.py
View File

@ -50,7 +50,7 @@ def decode_video_frames(
""" """
if backend == "torchcodec": if backend == "torchcodec":
return decode_video_frames_torchcodec(video_path, timestamps, tolerance_s) return decode_video_frames_torchcodec(video_path, timestamps, tolerance_s)
elif backend == "pyav": elif backend in ["pyav", "video_reader"]:
return decode_video_frames_torchvision(video_path, timestamps, tolerance_s, backend) return decode_video_frames_torchvision(video_path, timestamps, tolerance_s, backend)
else: else:
raise ValueError(f"Unsupported video backend: {backend}") raise ValueError(f"Unsupported video backend: {backend}")