predict_video

lightning_pose.utils.predictions.predict_video(video_file: str, model: Model, output_pred_file: str | None = None, progress_file: Path | None = None) → DataFrame[source]
lightning_pose.utils.predictions.predict_video(video_file: list[str], model: Model, output_pred_file: list[str] | None = None, progress_file: Path | None = None) → list[DataFrame]
Parameters:

  • video_file – Predict on a video, or for true multiview models, a list of videos (order: 1-1 correspondence with cfg.data.view_names).

  • model – The model to predict with.

  • output_pred_file – (optional) File to save predictions in. For multiview, a list of files (1-1 correspondance to cfg.data.view_names).