lightning_pose

lightning_pose.callbacks Module

Classes

AnnealWeight	Callback to change weight value during training.
UnfreezeBackbone	Callback that ramps up the backbone learning rate from 0 to upsampling_lr on unfreeze_epoch or unfreeze_step.

lightning_pose.metrics Module

Functions

pixel_error(keypoints_true, keypoints_pred)	Root mean square error between true and predicted keypoints.
temporal_norm(keypoints_pred)	Norm of difference between keypoints on successive time bins.
pca_singleview_reprojection_error(...)	PCA reprojection error.
pca_multiview_reprojection_error(...)	PCA reprojection error.