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

add convert stats in parallel using multiple thread and decord video backend.

This commit is contained in:
ChopinChen 2025-04-09 15:00:35 +08:00
parent d8f53d93c0
commit dcd0f5c519
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -24,6 +24,7 @@ from pathlib import Path
from typing import Any, ClassVar from typing import Any, ClassVar
import decord import decord
import numpy as np
import pyarrow as pa import pyarrow as pa
import torch import torch
import torchvision import torchvision
@ -249,7 +250,7 @@ def decode_video_frames_torchcodec(
def decode_video_frames_decord( def decode_video_frames_decord(
video_path: Path | str, video_path: Path | str,
timestamps: list[float], timestamps: list[float],
) -> torch.Tensor:: ) -> torch.Tensor:
video_path = str(video_path) video_path = str(video_path)
vr = decord.VideoReader(video_path) vr = decord.VideoReader(video_path)
num_frames = len(vr) num_frames = len(vr)