File size: 898 Bytes
fe64bad |
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 |
import numpy as np
import imageio
def load_video(filename):
vid = imageio.get_reader(filename, 'ffmpeg')
fps = vid.get_meta_data()['fps']
nframes = vid.count_frames()
return vid, fps, nframes
class SaveVideo:
def __init__(self, outname, fps):
self.outname = outname
self.fps = fps
def __enter__(self):
self.writter = imageio.get_writer(self.outname,
format='FFMPEG',
fps=self.fps)
return self
def __exit__(self, exc_type, exc_value, exc_traceback):
self.writter.close()
def __iadd__(self, data):
if np.max(data) <= 1:
data = np.array(255*data, dtype=np.uint8)
else:
data = np.array(data, dtype=np.uint8)
self.writter.append_data(data)
return self
|