administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-16 16:15:22 +03:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Manage faces properly

Browse Source
This commit is contained in:
Nicolas Mowen 2024-10-23 08:08:52 -06:00
parent b34a51903f
commit d30771c6f5
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
frigate/embeddings/maintainer.py
View File

@ -185,6 +185,9 @@ class EmbeddingMaintainer(threading.Thread):
event_id, camera, updated_db = ended event_id, camera, updated_db = ended
camera_config = self.config.cameras[camera] camera_config = self.config.cameras[camera]
if event_id in self.detected_faces:
self.detected_faces.pop(event_id)
if updated_db: if updated_db:
try: try:
event: Event = Event.get(Event.id == event_id) event: Event = Event.get(Event.id == event_id)
@ -368,6 +371,11 @@ class EmbeddingMaintainer(threading.Thread):
avg_score += score avg_score += score
avg_score = avg_score / REQUIRED_FACES avg_score = avg_score / REQUIRED_FACES
if id in self.detected_faces and avg_score <= self.detected_faces[id]:
logger.debug("Detected face does not score higher than previous face.")
return None
self.detected_faces[id] = avg_score self.detected_faces[id] = avg_score
requests.post( requests.post(
f"{FRIGATE_LOCALHOST}/api/events/{id}/sub_label", f"{FRIGATE_LOCALHOST}/api/events/{id}/sub_label",