decord_video_reader_frame_extraction_and_metadata.py

1import decord
2from decord import VideoReader
3from decord import cpu, gpu
4
5# 1. Setup - Use CPU for frame extraction
6ctx = cpu(0)
7
8# 2. Load the video
9# Replace 'video_file.mp4' with the path to your video
10video_path = 'video_file.mp4'
11vr = VideoReader(video_path, ctx=ctx)
12
13# 3. Access Metadata
14print(f"Total number of frames: {len(vr)}")
15
16# 4. Access individual frames
17# Get the first frame (frame 0)
18frame = vr[0]
19print(f"Frame shape: {frame.shape}") # (Height, Width, Channels)
20
21# 5. Access a range of frames
22# Get frames from index 10 to 14
23frames = vr.get_batch([10, 11, 12, 13, 14])
24print(f"Batch shape: {frames.shape}") # (NumFrames, Height, Width, Channels)
25
26# 6. Convert to NumPy array
27# decord frames are returned as decord.ndarray, convert to numpy for processing
28numpy_frame = frame.asnumpy()