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.

PatchMasking

Callback to apply curriculum patch masking during training.

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.