dgs.models.similarity.pose_similarity.ObjectKeypointSimilarity.get_data

ObjectKeypointSimilarity.get_data(ds: State) → torch.Tensor

Given a State, compute the detected / predicted key points with shape [B1 x J x 2|3] and the areas of the respective ground-truth bounding-boxes with shape [B1].