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

bird classification dynamic config

Browse Source
This commit is contained in:
Josh Hawkins 2026-02-25 08:27:49 -06:00
parent bf2741e979
commit 79242d49ca
2 changed files with 18 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
frigate/embeddings/maintainer.py
View File

@ -99,6 +99,9 @@ class EmbeddingMaintainer(threading.Thread):
self.classification_config_subscriber = ConfigSubscriber( self.classification_config_subscriber = ConfigSubscriber(
"config/classification/custom/" "config/classification/custom/"
) )
self.bird_classification_config_subscriber = ConfigSubscriber(
"config/classification", exact=True
)
self.face_recognition_config_subscriber = ConfigSubscriber( self.face_recognition_config_subscriber = ConfigSubscriber(
"config/face_recognition", exact=True "config/face_recognition", exact=True
) )
@ -277,6 +280,7 @@ class EmbeddingMaintainer(threading.Thread):
while not self.stop_event.is_set(): while not self.stop_event.is_set():
self.config_updater.check_for_updates() self.config_updater.check_for_updates()
self._check_classification_config_updates() self._check_classification_config_updates()
self._check_bird_classification_config_updates()
self._check_face_recognition_config_updates() self._check_face_recognition_config_updates()
self._check_lpr_config_updates() self._check_lpr_config_updates()
self._process_requests() self._process_requests()
@ -290,6 +294,7 @@ class EmbeddingMaintainer(threading.Thread):
self.config_updater.stop() self.config_updater.stop()
self.classification_config_subscriber.stop() self.classification_config_subscriber.stop()
self.bird_classification_config_subscriber.stop()
self.face_recognition_config_subscriber.stop() self.face_recognition_config_subscriber.stop()
self.lpr_config_subscriber.stop() self.lpr_config_subscriber.stop()
self.event_subscriber.stop() self.event_subscriber.stop()
@ -364,6 +369,18 @@ class EmbeddingMaintainer(threading.Thread):
f"Added classification processor for model: {model_name} (type: {type(processor).__name__})" f"Added classification processor for model: {model_name} (type: {type(processor).__name__})"
) )
def _check_bird_classification_config_updates(self) -> None:
"""Check for bird classification config updates."""
topic, classification_config = (
self.bird_classification_config_subscriber.check_for_update()
)
if topic is None:
return
self.config.classification = classification_config
logger.debug("Applied dynamic bird classification config update")
def _check_face_recognition_config_updates(self) -> None: def _check_face_recognition_config_updates(self) -> None:
"""Check for face recognition config updates.""" """Check for face recognition config updates."""
topic, face_config = self.face_recognition_config_subscriber.check_for_update() topic, face_config = self.face_recognition_config_subscriber.check_for_update()

2
web/src/components/config-form/section-configs/classification.ts
View File

@ -3,7 +3,7 @@ import type { SectionConfigOverrides } from "./types";
const classification: SectionConfigOverrides = { const classification: SectionConfigOverrides = {
base: { base: {
sectionDocs: "/configuration/custom_classification/object_classification", sectionDocs: "/configuration/custom_classification/object_classification",
restartRequired: ["bird.enabled", "bird.threshold"], restartRequired: ["bird.enabled"],
hiddenFields: ["custom"], hiddenFields: ["custom"],
advancedFields: [], advancedFields: [],
}, },