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

Cleanup

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
frigate/util/object.py
View File

@ -154,7 +154,7 @@ def get_region_from_grid(
return new 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_name = obj[0]
object_score = obj[1] object_score = obj[1]
object_box = obj[2] object_box = obj[2]

9
frigate/video.py
View File

@ -33,7 +33,12 @@ from frigate.util.image import (
intersection, intersection,
intersection_over_union, 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 from frigate.util.services import listen
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@ -525,7 +530,7 @@ def detect(
region, region,
) )
# apply object filters # apply object filters
if filtered(det, objects_to_track, object_filters): if is_object_filtered(det, objects_to_track, object_filters):
continue continue
detections.append(det) detections.append(det)
return detections return detections