File size: 1,029 Bytes
5db43ff |
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 |
import os
import cv2
import numpy as np
import imageio
class Image2VideoWriter():
def __init__(self,quality=5):
self.image_list = []
self.quality=quality
def append(self,image,isRGB=False):
if not isRGB:
image = image[:, :, [2,1,0]]
self.image_list.append(image)
def make_video(self,outvid=None, fps=5):
writer = imageio.get_writer(outvid, fps=fps, codec='libx264',quality=self.quality)
for image in self.image_list:
writer.append_data(image)
writer.close()
class StreamImage2Video():
def __init__(self,outvid,fps=30,quality=5):
self.outvid = outvid
self.quality=quality
self.fps = fps
self.vid=imageio.get_writer(outvid, fps=fps, codec='libx264',quality=self.quality)
self.image_list = []
def append(self,image,isRGB=False):
if not isRGB:
image = image[:, :, [2,1,0]]
self.vid.append_data(image)
def make_video(self):
self.vid.close()
|