From 419d5006c648e05697dbd24a2da9d1663f25ed68 Mon Sep 17 00:00:00 2001 From: Josh Hawkins <32435876+hawkeye217@users.noreply.github.com> Date: Sat, 14 Oct 2023 12:58:20 -0500 Subject: [PATCH] fix velocity bug --- frigate/ptz/autotrack.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/frigate/ptz/autotrack.py b/frigate/ptz/autotrack.py index 9067098d8..5cce42949 100644 --- a/frigate/ptz/autotrack.py +++ b/frigate/ptz/autotrack.py @@ -524,16 +524,16 @@ class PtzAutoTracker: def _get_valid_velocity(self, camera, obj): # returns a tuple and euclidean distance if the estimated velocity is valid - # if invalid, returns (0, 0, 0, 0) and -1 + # if invalid, returns [0, 0] and -1 camera_config = self.config.cameras[camera] camera_width = camera_config.frame_shape[1] camera_height = camera_config.frame_shape[0] camera_fps = camera_config.detect.fps velocities = obj.obj_data["estimate_velocity"] + diff = np.abs(velocities[1] - velocities[0]) - vel_mags = np.linalg.norm(np.diff(velocities, axis=0), axis=1) - if np.any(vel_mags > 8): + if (diff > 8).any(): # invalid velocity return np.zeros(2), -1