Spaces:
Runtime error
Runtime error
File size: 1,122 Bytes
09295f0 db0445e 09295f0 db0445e |
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 |
from pytube import YouTube
import numpy as np
from decord import VideoReader, cpu
import imageio
def download_youtube_video(url: str):
yt = YouTube(url)
streams = yt.streams.filter(file_extension="mp4")
file_path = streams[0].download()
return file_path
def sample_frame_indices(clip_len, frame_sample_rate):
converted_len = int(clip_len * frame_sample_rate)
start_idx = 0
end_idx = converted_len
indices = np.linspace(start_idx, end_idx, num=clip_len)
indices = np.clip(indices, start_idx, end_idx - 1).astype(np.int64)
return indices
def sample_frames_from_video_file(file_path: str, num_frames: int = 16):
videoreader = VideoReader(file_path, num_threads=1, ctx=cpu(0))
# sample frames
videoreader.seek(0)
indices = sample_frame_indices(clip_len=num_frames, frame_sample_rate=4)
frames = videoreader.get_batch(indices).asnumpy()
return frames
def convert_frames_to_gif(frames):
SAVE_PATH = "frames.gif"
converted_frames = frames.astype(np.uint8)
imageio.mimsave(SAVE_PATH, converted_frames, duration=125) # 8fps
return SAVE_PATH
|