File size: 516 Bytes
0ea5594 | 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 | import os
import skvideo.io
def _make_dir(filename):
folder = os.path.dirname(filename)
if not os.path.exists(folder):
os.makedirs(folder)
def save_video(filename, video_frames, fps=60, video_format="mp4"):
assert fps == int(fps), fps
_make_dir(filename)
skvideo.io.vwrite(
filename,
video_frames,
inputdict={
"-r": str(int(fps)),
},
outputdict={
"-f": video_format,
"-pix_fmt": "yuv420p",
},
)
|