Spaces:
Runtime error
Runtime error
File size: 1,403 Bytes
5fb8331 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 |
# Importing the requirements
from PIL import Image
from decord import VideoReader, cpu
# Maximum number of frames to use
MAX_NUM_FRAMES = 64 # If CUDA OOM, set a smaller number
def encode_video(video_path):
"""
Encodes a video file into a list of frames.
Args:
video_path (str): The path to the video file.
Returns:
list: A list of frames, where each frame is represented as an Image object.
"""
def uniform_sample(l, n):
"""
Uniformly samples elements from a list.
Args:
- l (list): The input list.
- n (int): The number of elements to sample.
Returns:
list: A list of sampled elements.
"""
gap = len(l) / n
idxs = [int(i * gap + gap / 2) for i in range(n)]
return [l[i] for i in idxs]
# Read the video file and sample frames
vr = VideoReader(video_path, ctx=cpu(0))
sample_fps = round(vr.get_avg_fps() / 1) # FPS
frame_idx = [i for i in range(0, len(vr), sample_fps)]
# Uniformly sample frames if the number of frames is too large
if len(frame_idx) > MAX_NUM_FRAMES:
frame_idx = uniform_sample(frame_idx, MAX_NUM_FRAMES)
# Extract frames from the video
frames = vr.get_batch(frame_idx).asnumpy()
frames = [Image.fromarray(v.astype("uint8")) for v in frames]
# Return video frames
return frames
|