Spartan/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2024-11-21 16:38:24 -06:00
Code Issues Packages Projects Releases Wiki Activity

Apply zone filter before inertia (#6854)

Browse Source 
* Apply zone filter before intertia

* Formatting
This commit is contained in:
Nicolas Mowen 2023-06-30 06:15:55 -06:00 committed by GitHub
parent bd17e7d824
commit d2a2643cd6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
frigate/object_processing.py
View File

@ -188,15 +188,14 @@ class TrackedObject:
zone_score = self.zone_presence.get(name, 0) zone_score = self.zone_presence.get(name, 0)
# check if the object is in the zone # check if the object is in the zone
if cv2.pointPolygonTest(contour, bottom_center, False) >= 0: if cv2.pointPolygonTest(contour, bottom_center, False) >= 0:
self.zone_presence[name] = zone_score + 1 # if the object passed the filters once, dont apply again
if name in self.current_zones or not zone_filtered(self, zone.filters):
self.zone_presence[name] = zone_score + 1
# an object is only considered present in a zone if it has a zone inertia of 3+ # an object is only considered present in a zone if it has a zone inertia of 3+
if zone_score >= zone.inertia: if zone_score >= zone.inertia:
# if the object passed the filters once, dont apply again
if name in self.current_zones or not zone_filtered(
self, zone.filters
):
current_zones.append(name) current_zones.append(name)
if name not in self.entered_zones: if name not in self.entered_zones:
self.entered_zones.append(name) self.entered_zones.append(name)
else: else: