decord_video_reader_random_frame_access_ndarray.py

1import decord
2from decord import VideoReader
3from decord import cpu, gpu
4
5# Load a video file
6# You can specify the device (cpu(0) or gpu(0))
7vr = VideoReader('your_video_file.mp4', ctx=cpu(0))
8
9# 1. Get the total number of frames
10print('Total frames:', len(vr))
11
12# 2. Access a single frame by index
13# Returns a decord.nd.NDArray
14frame = vr[0]
15print('Frame shape:', frame.shape)
16
17# 3. Access multiple frames at once
18# Returns a decord.nd.NDArray with shape (batch, height, width, channel)
19frames = vr.get_batch([0, 10, 20])
20print('Batch shape:', frames.shape)
21
22# 4. Convert decord NDArray to numpy array if needed
23numpy_frame = frame.asnumpy()