undo_affine_transform_batch

lightning_pose.data.utils.undo_affine_transform_batch(keypoints_augmented: Float[torch.Tensor, 'seq_len num_keypointsx2'], transforms: Float[torch.Tensor, 'seq_len h w'] | Float[torch.Tensor, 'h w'] | Float[torch.Tensor, 'seq_len 1'] | Float[torch.Tensor, 1] | Float[torch.Tensor, 'num_views h w'] | Float[torch.Tensor, 'num_views 1 1'], is_multiview: bool = False) Float[torch.Tensor, 'seq_len num_keypointsx2'][source]

Potentially undo an affine transform given a tensor of keypoints and the tranform matrix.