Create videos from numpy arrays in a jupyter notebook
Packaged version of Alexander Mordvintsev's Colab Snippets
pip install notebook-video-writer
- Must have ffmpeg installed
import numpy as np
# optionally wrap with tqdm for progress bar
from tqdm import tqdm
from notebook_video_writer import VideoWriter
with VideoWriter(fps=40) as vw:
for i in tqdm(range(100)):
frame = np.random.rand(256,256,3)
vw.add(frame)This will immediately display the video inline below the cell:
An mp4 file is also saved to the current directory. Default name is _autoplay.mp4, but can be set with name keyword arg.