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

Formatting

Browse Source
This commit is contained in:
Nick Mowen 2023-05-24 10:54:46 -06:00
parent 65be2dc898
commit 80fa251ad3
1 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
frigate/track/norfair_tracker.py
View File

@ -27,13 +27,17 @@ def frigate_distance(detection: Detection, tracked_object) -> float:
# calculate euclidean distance and average # calculate euclidean distance and average
distance = np.linalg.norm(difference, axis=1).mean() distance = np.linalg.norm(difference, axis=1).mean()
detected_area = (detection.points[1][0] - detection.points[0][0]) * (detection.points[1][1] - detection.points[0][1]) detected_area = (detection.points[1][0] - detection.points[0][0]) * (
estimated_area = (tracked_object.estimate[1][0] - tracked_object.estimate[0][0]) * (tracked_object.estimate[1][1] - tracked_object.estimate[0][1]) detection.points[1][1] - detection.points[0][1]
)
estimated_area = (tracked_object.estimate[1][0] - tracked_object.estimate[0][0]) * (
tracked_object.estimate[1][1] - tracked_object.estimate[0][1]
)
# calculate the factor of area increase / decrease between detected and estimated bbox # calculate the factor of area increase / decrease between detected and estimated bbox
area_factor = detected_area / estimated_area area_factor = detected_area / estimated_area
# account for cases where object is smaller than estimated # account for cases where object is smaller than estimated
if area_factor < 1: if area_factor < 1:
area_factor = 1 / area_factor area_factor = 1 / area_factor