From 77a1faa7defaf16f0ef41db98c959c45f9cb408a Mon Sep 17 00:00:00 2001 From: arjun-234 <103405661+arjun-234@users.noreply.github.com> Date: Thu, 17 Aug 2023 15:45:53 +0530 Subject: [PATCH 1/2] refactor: multi_predict() In this version, I have modified the code by combining three loops into a single loop within the same method. This enhancement has resulted in improved efficiency. --- supervision/tracker/byte_tracker/core.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/supervision/tracker/byte_tracker/core.py b/supervision/tracker/byte_tracker/core.py index 2f4e99a1..30c02021 100644 --- a/supervision/tracker/byte_tracker/core.py +++ b/supervision/tracker/byte_tracker/core.py @@ -33,13 +33,16 @@ class STrack(BaseTrack): @staticmethod def multi_predict(stracks): if len(stracks) > 0: - multi_mean = np.asarray([st.mean.copy() for st in stracks]) - multi_covariance = np.asarray([st.covariance for st in stracks]) + multi_mean = [] + multi_covariance = [] for i, st in enumerate(stracks): + multi_mean.append(st.mean.copy()) + multi_covariance.append(st.covariance) if st.state != TrackState.Tracked: multi_mean[i][7] = 0 + multi_mean, multi_covariance = STrack.shared_kalman.multi_predict( - multi_mean, multi_covariance + np.asarray(multi_mean), np.asarray(multi_covariance) ) for i, (mean, cov) in enumerate(zip(multi_mean, multi_covariance)): stracks[i].mean = mean From cd8541f9484ef309894eb01450caadd9dfb4b313 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Thu, 17 Aug 2023 10:18:18 +0000 Subject: [PATCH 2/2] =?UTF-8?q?fix(pre=5Fcommit):=20=F0=9F=8E=A8=20auto=20?= =?UTF-8?q?format=20pre-commit=20hooks?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- supervision/tracker/byte_tracker/core.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/supervision/tracker/byte_tracker/core.py b/supervision/tracker/byte_tracker/core.py index 30c02021..466b587e 100644 --- a/supervision/tracker/byte_tracker/core.py +++ b/supervision/tracker/byte_tracker/core.py @@ -40,7 +40,7 @@ class STrack(BaseTrack): multi_covariance.append(st.covariance) if st.state != TrackState.Tracked: multi_mean[i][7] = 0 - + multi_mean, multi_covariance = STrack.shared_kalman.multi_predict( np.asarray(multi_mean), np.asarray(multi_covariance) )