administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-07 03:35:26 +03:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

numpy zeros dimension

Browse Source
This commit is contained in:
Josh Hawkins 2023-10-11 18:50:07 -05:00
parent b7748feb6c
commit 80ffc690ed
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
frigate/ptz/autotrack.py
View File

@ -535,7 +535,7 @@ class PtzAutoTracker:
vel_mags = np.linalg.norm(np.diff(velocities, axis=0), axis=1) vel_mags = np.linalg.norm(np.diff(velocities, axis=0), axis=1)
if np.any(vel_mags > 8): if np.any(vel_mags > 8):
# invalid velocity # invalid velocity
return np.zeros((1, 2)), -1 return np.zeros(2), -1
average_velocity = np.mean(velocities, axis=0) average_velocity = np.mean(velocities, axis=0)
@ -543,14 +543,14 @@ class PtzAutoTracker:
velocity_distance = np.linalg.norm(velocities[0] - velocities[1]) velocity_distance = np.linalg.norm(velocities[0] - velocities[1])
if ( if (
0 <= np.abs(average_velocity[0]) < (camera_width / camera_fps / 2) 0 <= abs(average_velocity[0]) < (camera_width / camera_fps / 2)
and 0 <= np.abs(average_velocity[1]) < (camera_height / camera_fps / 2) and 0 <= abs(average_velocity[1]) < (camera_height / camera_fps / 2)
and velocity_distance <= 10 and velocity_distance <= 10
): ):
return average_velocity, velocity_distance return average_velocity, velocity_distance
else: else:
# invalid velocity # invalid velocity
return np.zeros((1, 2)), -1 return np.zeros(2), -1
def _below_distance_threshold(self, camera, obj): def _below_distance_threshold(self, camera, obj):
# Returns true if Euclidean distance from object to center of frame is # Returns true if Euclidean distance from object to center of frame is