diff --git a/pose_evaluation/metrics/distance_metric.py b/pose_evaluation/metrics/distance_metric.py index db41183..31b9a68 100644 --- a/pose_evaluation/metrics/distance_metric.py +++ b/pose_evaluation/metrics/distance_metric.py @@ -25,8 +25,8 @@ def score(self, hypothesis: Pose, reference: Pose) -> float: # Calculate the error error = arrays[0] - arrays[1] - # for l2, we need to calculate the error for each point - if self.kind == "l2": + # for l2/euclidean, we need to calculate the error for each point + if self.kind == "euclidean": # the last dimension is the 3D coordinates error = ma.power(error, 2) error = error.sum(axis=-1)