torchcodec_video_decoder_metadata_and_frame_extraction.py

1import torch
2from torchcodec.decoders import VideoDecoder
3
4# Path to your video file
5video_path = "video.mp4"
6
7# Create a VideoDecoder instance
8decoder = VideoDecoder(video_path)
9
10# Access metadata
11print(f"Metadata: {decoder.metadata}")
12
13# Extract the first frame (index 0)
14# This returns a Frame object containing the data as a torch.Tensor
15frame = decoder.get_frame_at_index(0)
16
17# The frame data is a tensor of shape (C, H, W)
18print(f"Frame tensor shape: {frame.data.shape}")
19print(f"Frame presentation timestamp: {frame.pts_seconds}")