From 7265848f9463fbeaf9df4b8f48f3ecad999a6ec5 Mon Sep 17 00:00:00 2001 From: Nick Mowen Date: Mon, 16 Oct 2023 16:16:58 -0600 Subject: [PATCH] Cleanup --- frigate/util/object.py | 2 +- frigate/video.py | 9 +++++++-- 2 files changed, 8 insertions(+), 3 deletions(-) diff --git a/frigate/util/object.py b/frigate/util/object.py index 825e7d53c..d90896d7a 100644 --- a/frigate/util/object.py +++ b/frigate/util/object.py @@ -154,7 +154,7 @@ def get_region_from_grid( return new -def filtered(obj, objects_to_track, object_filters): +def is_object_filtered(obj, objects_to_track, object_filters): object_name = obj[0] object_score = obj[1] object_box = obj[2] diff --git a/frigate/video.py b/frigate/video.py index b2cea0980..b5735fcf6 100755 --- a/frigate/video.py +++ b/frigate/video.py @@ -33,7 +33,12 @@ from frigate.util.image import ( intersection, intersection_over_union, ) -from frigate.util.object import get_cluster_region_from_grid +from frigate.util.object import ( + create_tensor_input, + get_cluster_region_from_grid, + get_min_region_size, + is_object_filtered, +) from frigate.util.services import listen logger = logging.getLogger(__name__) @@ -525,7 +530,7 @@ def detect( region, ) # apply object filters - if filtered(det, objects_to_track, object_filters): + if is_object_filtered(det, objects_to_track, object_filters): continue detections.append(det) return detections