lightning_pose.models.heads

lightning_pose.models.heads Package

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.