lightning_pose.models.heads

lightning_pose.models.heads Package

Prediction head classes, re-exported at the package level.

Classes

HeatmapHead

Simple deconvolution head that converts 2D feature maps to per-keypoint heatmaps.

HeatmapMHCRNNHead

Multi-head convolutional recurrent neural network head.

LinearRegressionHead

Linear regression head that converts 2D feature maps to a vector of (x, y) coordinates.