lisbet.utils#

LISBET

Functions

export_pose_video(frames, filename[, fps])

Export a sequence of frames (numpy arrays, shape [N, H, W, 3]) to mp4 using imageio.

permute_predictions(predictions, labels)

lisbet.utils.permute_predictions(predictions, labels)[source]#
lisbet.utils.export_pose_video(frames, filename, fps=30)[source]#

Export a sequence of frames (numpy arrays, shape [N, H, W, 3]) to mp4 using imageio.

Args:

frames: numpy array of shape (num_frames, height, width, 3), dtype=uint8 filename: output mp4 file path fps: frames per second