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.