lightning_pose.models

lightning_pose.models Package

Classes

HeatmapTracker	Base model that produces heatmaps of keypoints from images.
SemiSupervisedHeatmapTracker	Model produces heatmaps of keypoints from labeled/unlabeled images.
HeatmapTrackerMHCRNN	Multi-headed Convolutional RNN network that handles context frames.
SemiSupervisedHeatmapTrackerMHCRNN	Model produces heatmaps of keypoints from labeled/unlabeled images.
HeatmapTrackerMultiviewTransformer	Transformer network that handles multi-view datasets.
SemiSupervisedHeatmapTrackerMultiviewTransformer	Semi-supervised HeatmapTrackerMultiviewTransformer that supports unsupervised losses.
RegressionTracker	Base model that produces (x, y) predictions of keypoints from images.
SemiSupervisedRegressionTracker	Model produces vectors of keypoints from labeled/unlabeled images.

Subpackages

• lightning_pose.models.heads