create_labeled_video

lightning_pose.utils.predictions.create_labeled_video(clip: VideoFileClip, xs_arr: ndarray, ys_arr: ndarray, mask_array: ndarray | None = None, dotsize: int = 4, colormap: str | None = 'cool', fps: float | None = None, output_video_path: str = 'movie.mp4', start_time: float = 0.0) → None

Helper function for creating annotated videos. Args

clip xs_arr: shape T x n_joints ys_arr: shape T x n_joints mask_array: shape T x n_joints; timepoints/joints with a False entry will not be plotted dotsize: size of marker dot on labeled video colormap: matplotlib color map for markers fps: None to default to fps of original video output_video_path: video file name start_time: time (in seconds) of video start