Shortcuts

FrameBatch

class torchcodec.FrameBatch(data: Tensor, pts_seconds: Tensor, duration_seconds: Tensor)[source]

多个视频帧及其相关的元数据。

data 张量通常对于帧序列是4D的(NHWC或NCHW),或者对于片段序列是5D的,如采样器返回的那样。当data是4D(相应地,5D)时,pts_secondsduration_seconds张量是1D(相应地,2D)的。

注意

pts_secondsduration_seconds 张量总是返回在 CPU 上,即使 data 在 GPU 上。

使用 FrameBatch 的示例:

使用VideoDecoder解码视频

Decoding a video with VideoDecoder

如何采样视频片段

How to sample video clips
data: Tensor

帧数据(torch.Tensor 类型为 uint8)。

duration_seconds: Tensor

帧的持续时间,以秒为单位(torch.Tensor 的浮点数)。

pts_seconds: Tensor

帧的pts,以秒为单位(torch.Tensor 浮点数)。