lightning_pose.models

lightning_pose.models.base Module

Base class for backbone that acts as a feature extractor.

Functions

normalized_to_bbox(keypoints, bbox)
convert_bbox_coords(batch_dict, ...)	Transform keypoints from bbox coordinates to absolute frame coordinates.
get_context_from_sequence(img_seq, ...)

Classes

BaseFeatureExtractor	Object that contains the base resnet feature extractor.
BaseSupervisedTracker	Base class for supervised trackers.
SemiSupervisedTrackerMixin	Mixin class providing training step function for semi-supervised models.

lightning_pose.models.heatmap_tracker Module

Models that produce heatmaps of keypoints from images.

Classes

HeatmapTracker	Base model that produces heatmaps of keypoints from images.
SemiSupervisedHeatmapTracker	Model produces heatmaps of keypoints from labeled/unlabeled images.

lightning_pose.models.heatmap_tracker_mhcrnn Module

Models that produce heatmaps of keypoints from images.

Classes

HeatmapTrackerMHCRNN	Multi-headed Convolutional RNN network that handles context frames.
SemiSupervisedHeatmapTrackerMHCRNN	Model produces heatmaps of keypoints from labeled/unlabeled images.

lightning_pose.models.regression_tracker Module

Models that produce (x, y) coordinates of keypoints from images.

Classes

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.backbones
• lightning_pose.models.heads