This commit is contained in:
Nick Mowen 2023-10-16 16:16:58 -06:00
parent b72d9a49b2
commit 7265848f94
2 changed files with 8 additions and 3 deletions

View File

@ -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]

View File

@ -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