get_model_class

lightning_pose.utils.predictions.get_model_class(map_type: str, semi_supervised: bool) → Type[HeatmapTracker | SemiSupervisedHeatmapTracker | HeatmapTrackerMHCRNN | SemiSupervisedHeatmapTrackerMHCRNN | HeatmapTrackerMultiviewTransformer | SemiSupervisedHeatmapTrackerMultiviewTransformer | RegressionTracker | SemiSupervisedRegressionTracker][source]

[summary]

Parameters:

  • map_type (str) – β€œregression” | β€œheatmap”

  • semi_supervised (bool) – True if you want to use unlabeled videos

Returns:

a ptl model class to be initialized outside of this function.