make_ae_reconstruction_movie_wrapper

behavenet.plotting.ae_utils.make_ae_reconstruction_movie_wrapper(hparams, save_file, trial=None, sess_idx=0, version='best', include_linear=False, max_frames=400, frame_rate=15)[source]

Produce movie with original video, reconstructed video, and residual.

This is a high-level function that loads the model described in the hparams dictionary and produces the necessary predicted video frames.

Parameters:

  • hparams (dict) – needs to contain enough information to specify an autoencoder

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

  • trial (int, optional) – if NoneType, use first test trial

  • sess_idx (int, optional) – session index into data generator

  • version (str or int, optional) – test tube model version

  • include_linear (bool, optional) – include reconstruction from corresponding linear ae (i.e. ame number of latents)

  • max_frames (int, optional) – maximum number of frames to animate from a trial

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