You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
44 lines
1.1 KiB
44 lines
1.1 KiB
1 year ago
|
import yaml
|
||
|
|
import pytube
|
||
|
|
import tempfile
|
||
|
|
import imageio
|
||
|
|
|
||
|
|
from pathlib import Path
|
||
|
|
from functools import lru_cache
|
||
|
|
import moviepy.editor as mpy
|
||
|
|
|
||
|
|
|
||
|
|
FILE = 'data/input/samples/exercises.yaml'
|
||
|
|
|
||
|
|
|
||
|
|
@lru_cache(5)
|
||
|
|
def get_video(url):
|
||
|
|
output_dir = tempfile.mktemp()
|
||
|
|
video = pytube.YouTube(url)
|
||
|
|
video.streams.filter(res="360p").first().download(filename=output_dir)
|
||
|
|
return output_dir
|
||
|
|
|
||
|
|
def create_gif_from_video_and_timestamps(video_path, start, end):
|
||
|
|
video = mpy.VideoFileClip(video_path)
|
||
|
|
clip: mpy.VideoFileClip = video.subclip(start, end)
|
||
|
|
output_dir = tempfile.mktemp() + '.mp4'
|
||
|
|
clip.set_fps(2)
|
||
|
|
clip.without_audio().write_videofile(output_dir)
|
||
|
|
return output_dir
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
file = open(FILE)
|
||
|
|
exercises = yaml.safe_load(file)
|
||
|
|
for ex in exercises['exercises']:
|
||
|
|
name = ex['name']
|
||
|
|
video_url = ex['video']
|
||
|
|
start = ex['start']
|
||
|
|
end = ex['end']
|
||
|
|
|
||
|
|
video_path = get_video(video_url)
|
||
|
|
gif = create_gif_from_video_and_timestamps(video_path, start, end)
|
||
|
|
ex['gif_path'] = gif
|
||
|
|
|
||
|
|
print(gif)
|