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

fix numpy shapes

Browse Source
This commit is contained in:
Josh Hawkins 2023-10-11 18:35:11 -05:00
parent 0825458bcb
commit b7748feb6c
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

@ -532,10 +532,10 @@ class PtzAutoTracker:
velocities = obj.obj_data["estimate_velocity"] velocities = obj.obj_data["estimate_velocity"]
vel_mags = np.linalg.norm(velocities, 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((2, 2)), -1 return np.zeros((1, 2)), -1
average_velocity = np.mean(velocities, axis=0) average_velocity = np.mean(velocities, axis=0)
@ -550,7 +550,7 @@ class PtzAutoTracker:
return average_velocity, velocity_distance return average_velocity, velocity_distance
else: else:
# invalid velocity # invalid velocity
return np.zeros((2, 2)), -1 return np.zeros((1, 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
@ -676,7 +676,7 @@ class PtzAutoTracker:
f"{camera}: Zoom test: below dimension threshold: {below_dimension_threshold} width: {(bb_right - bb_left) / camera_width}, height: { (bb_bottom - bb_top) / camera_height}" f"{camera}: Zoom test: below dimension threshold: {below_dimension_threshold} width: {(bb_right - bb_left) / camera_width}, height: { (bb_bottom - bb_top) / camera_height}"
) )
logger.debug( logger.debug(
f"{camera}: Zoom test: below velocity threshold: {below_velocity_threshold} velocity x: {abs(np.mean(average_velocity[:, 0]))}, x threshold: {velocity_threshold_x}, velocity y: {abs(np.mean(average_velocity[:, 1]))}, y threshold: {velocity_threshold_y}" f"{camera}: Zoom test: below velocity threshold: {below_velocity_threshold} velocity x: {abs(average_velocity[0])}, x threshold: {velocity_threshold_x}, velocity y: {abs(average_velocity[1])}, y threshold: {velocity_threshold_y}"
) )
logger.debug(f"{camera}: Zoom test: at max zoom: {at_max_zoom}") logger.debug(f"{camera}: Zoom test: at max zoom: {at_max_zoom}")
logger.debug(f"{camera}: Zoom test: at min zoom: {at_min_zoom}") logger.debug(f"{camera}: Zoom test: at min zoom: {at_min_zoom}")