A match between a prediction and a ground truth assembly. The ground truth assembly should be None f the prediction was not matched to any GT, and the OKS should be 0. Attributes: prediction: A prediction made by a pose model. score: The confidence score for the predict
| 882 | |
| 883 | @dataclass |
| 884 | class MatchedPrediction: |
| 885 | """A match between a prediction and a ground truth assembly. |
| 886 | |
| 887 | The ground truth assembly should be None f the prediction was not matched to any GT, |
| 888 | and the OKS should be 0. |
| 889 | |
| 890 | Attributes: |
| 891 | prediction: A prediction made by a pose model. |
| 892 | score: The confidence score for the prediction. |
| 893 | ground_truth: If None, then this prediction is not matched to any ground truth |
| 894 | (this can happen when there are more predicted individuals than GT). |
| 895 | Otherwise, the ground truth assembly to which this prediction is matched. |
| 896 | oks: The OKS score between the prediction and the ground truth pose. |
| 897 | """ |
| 898 | |
| 899 | prediction: Assembly |
| 900 | score: float |
| 901 | ground_truth: Assembly | None |
| 902 | oks: float |
| 903 | |
| 904 | |
| 905 | def calc_object_keypoint_similarity( |