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