Spaces:
Runtime error
Runtime error
File size: 1,159 Bytes
ac7cda5 2089ecf ac7cda5 |
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 |
import imageio
import os
class VideoWriterByImageIO:
def __init__(self, video_path, fps=20, **kwargs):
video_format = kwargs.get("format", "mp4") # default is mp4 format
codec = kwargs.get("vcodec", "libx264") # default is libx264 encoding
quality = kwargs.get("quality") # video quality
pixelformat = kwargs.get("pixelformat", "yuv420p") # video pixel format
macro_block_size = kwargs.get("macro_block_size", 2)
ffmpeg_params = ["-crf", str(kwargs.get("crf", 18))]
os.makedirs(os.path.dirname(video_path), exist_ok=True)
writer = imageio.get_writer(
video_path,
fps=fps,
format=video_format,
codec=codec,
quality=quality,
ffmpeg_params=ffmpeg_params,
pixelformat=pixelformat,
macro_block_size=macro_block_size,
)
self.writer = writer
def __call__(self, img, fmt="bgr"):
if fmt == "bgr":
frame = img[..., ::-1]
else:
frame = img
self.writer.append_data(frame)
def close(self):
self.writer.close()
|