make_reconstruction_movie

behavenet.plotting.ae_utils.make_reconstruction_movie(ims, titles=None, n_rows=0, n_cols=0, save_file=None, frame_rate=15, dpi=100)[source]

Produce movie with original video and reconstructed videos.

ims and titles are corresponding lists; this data is plotted using a linear index, i.e. if n_rows = 2 and n_cols = 3 the image stack in ims[2] will be in the first row, second column; the image stack in ims[4] will be in the second row, first column. If ims[i] is empty, that grid location will be skipped.

Parameters:

  • ims (list of np.ndarray) – each list element is of shape (n_frames, n_channels, y_pix, x_pix)

  • titles (list of str, optional) – title for each panel

  • n_rows (int) – number of rows in video grid layout

  • n_cols (int) – number of columns in video grid layout

  • save_file (str, optional) – full save file (path and filename)

  • frame_rate (float, optional) – frame rate of saved movie

  • dpi (int, optional) – dpi of movie figure; controls resolution of titles