From 80ffc690edb6ff6bfda9941e9ace0728ed60216e Mon Sep 17 00:00:00 2001 From: Josh Hawkins <32435876+hawkeye217@users.noreply.github.com> Date: Wed, 11 Oct 2023 18:50:07 -0500 Subject: [PATCH] numpy zeros dimension --- frigate/ptz/autotrack.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/frigate/ptz/autotrack.py b/frigate/ptz/autotrack.py index 2256db23d..27747f1a5 100644 --- a/frigate/ptz/autotrack.py +++ b/frigate/ptz/autotrack.py @@ -535,7 +535,7 @@ class PtzAutoTracker: vel_mags = np.linalg.norm(np.diff(velocities, axis=0), axis=1) if np.any(vel_mags > 8): # invalid velocity - return np.zeros((1, 2)), -1 + return np.zeros(2), -1 average_velocity = np.mean(velocities, axis=0) @@ -543,14 +543,14 @@ class PtzAutoTracker: velocity_distance = np.linalg.norm(velocities[0] - velocities[1]) if ( - 0 <= np.abs(average_velocity[0]) < (camera_width / camera_fps / 2) - and 0 <= np.abs(average_velocity[1]) < (camera_height / camera_fps / 2) + 0 <= abs(average_velocity[0]) < (camera_width / camera_fps / 2) + and 0 <= abs(average_velocity[1]) < (camera_height / camera_fps / 2) and velocity_distance <= 10 ): return average_velocity, velocity_distance else: # invalid velocity - return np.zeros((1, 2)), -1 + return np.zeros(2), -1 def _below_distance_threshold(self, camera, obj): # Returns true if Euclidean distance from object to center of frame is