decord_videoreader_load_frames_as_numpy_arrays.py

1from decord import VideoReader
2from decord import cpu, gpu
3
4# Load the video using VideoReader
5# You can specify the context (cpu or gpu)
6vr = VideoReader('test_video.mp4', ctx=cpu(0))
7
8# 1. Get the length of the video (number of frames)
9print('Video length:', len(vr))
10
11# 2. Read a single frame using indexing
12frame = vr[0]
13print('Single frame shape:', frame.shape)
14
15# 3. Read multiple frames at once using get_batch
16# This returns a decord NDArray which can be converted to NumPy
17frame_indices = [1, 3, 5, 7, 9]
18frames = vr.get_batch(frame_indices).asnumpy()
19print('Batch frames shape:', frames.shape)