dgs.models.similarity.pose_similarity.ObjectKeypointSimilarity.get_target

ObjectKeypointSimilarity.get_target(ds: State) → tuple[torch.Tensor, torch.Tensor]

Given a State obtain the ground truth key points and the key-point-visibility. Both are tensors, the key points are a FloatTensor of shape [B2 x J x 2|3] and the visibility is a BoolTensor of shape [B2 x J].