export_predictions_and_labeled_video

lightning_pose.utils.predictions.export_predictions_and_labeled_video(video_file: str, cfg: DictConfig, prediction_csv_file: str, ckpt_file: str | None = None, trainer: Trainer | None = None, model: HeatmapTracker | SemiSupervisedHeatmapTracker | HeatmapTrackerMHCRNN | SemiSupervisedHeatmapTrackerMHCRNN | RegressionTracker | SemiSupervisedRegressionTracker | None = None, data_module: BaseDataModule | UnlabeledDataModule | None = None, labeled_mp4_file: str | None = None, save_heatmaps: bool | None = False) → DataFrame

Export predictions csv and a labeled video for a single video file.