From 66aa79e47a26b15625ec1477aae284204a686443 Mon Sep 17 00:00:00 2001 From: Jan Grewe Date: Fri, 7 Feb 2025 09:11:33 +0100 Subject: [PATCH] [skeleton] use numpy whereever possible ;) --- fixtracks/widgets/skeleton.py | 9 ++------- 1 file changed, 2 insertions(+), 7 deletions(-) diff --git a/fixtracks/widgets/skeleton.py b/fixtracks/widgets/skeleton.py index 6e7e80a..13c8f44 100644 --- a/fixtracks/widgets/skeleton.py +++ b/fixtracks/widgets/skeleton.py @@ -41,13 +41,8 @@ class Skeleton(QGraphicsRectItem): @property def length(self): - dist = 0.0 - for i in range(1, len(self.bodyaxis)): - si = self.bodyaxis[i-1] - ei = self.bodyaxis[i] - sk = self._keypoints[si, :] - ek = self._keypoints[ei, :] - dist += np.sqrt((ek[0] - sk[0])**2 + (ek[1] - sk[1])**2) + bodykps = self._keypoints[self.bodyaxis, :] + dist = np.sum(np.sqrt(np.sum(np.diff(bodykps, axis=0)**2, axis=1)), axis=0) return dist # def mousePressEvent(self, event):