hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

Update lerobot/common/datasets/video_utils.py

Browse Source 
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
1 changed files with 1 additions and 1 deletions
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":
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)
else:
raise ValueError(f"Unsupported video backend: {backend}")