Code Snippets for displaying image frames (NWHC format, numpy array) in Jupyter Notebook

Frames-to-GIF.py

import numpy as np
from matplotlib import pyplot as plt
from matplotlib import animation, rc
from IPython.core.display import display, HTML

%matplotlib inline
%load_ext autoreload
%autoreload 2

def display_frames_as_gif(frames):
    fig, ax = plt.subplots()
    patch = ax.imshow(frames[0])
    title_obj = ax.set_title(0)
    def animate(i):
        patch.set_data(frames[i])
        plt.setp(title_obj,color='k', text=('Time: %d' % (i)))
    
    anim = animation.FuncAnimation(fig, animate, range(len(frames)), interval=50)
    display(HTML(anim.to_html5_video()))
    plt.close()

Frames-to-mp4.py

import moviepy.editor as mpy
clip = mpy.ImageSequenceClip([(ob*255.).astype(np.uint8) for ob in obs],fps=30)
clip.write_videofile('video.mp4', verbose=False,ffmpeg_params=['-y'],progress_bar=False)