make_dlc_pandas_index

lightning_pose.utils.predictions.make_dlc_pandas_index(cfg: DictConfig | ListConfig, keypoint_names: list[str]) → MultiIndex[source]

Create a DLC-style three-level pandas MultiIndex for prediction DataFrames.

Parameters:

  • cfg – hydra config used to obtain the model type for the scorer level.

  • keypoint_names – list of body-part names.

Returns:

pd.MultiIndex with levels ["scorer", "bodyparts", "coords"] where coords are ["x", "y", "likelihood"].