From 6c6be1e0487d5a280305af25e4558396f2d444bd Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Fri, 23 May 2025 05:18:22 -0600 Subject: [PATCH] Add basic config for defining a teachable machine model --- frigate/config/classification.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/frigate/config/classification.py b/frigate/config/classification.py index 4c34f8ab3..dea991d92 100644 --- a/frigate/config/classification.py +++ b/frigate/config/classification.py @@ -34,10 +34,19 @@ class BirdClassificationConfig(FrigateBaseModel): ) +class TeachableMachineConfig(FrigateBaseModel): + enabled: bool = Field(default=True, title="Enable running the model.") + model_path: str = Field(title="Path to teachable machine tflite model.") + labelmap_path: str = Field(title="Path to teachable machine labelmap.") + + class ClassificationConfig(FrigateBaseModel): bird: BirdClassificationConfig = Field( default_factory=BirdClassificationConfig, title="Bird classification config." ) + teachable_machine: Dict[str, TeachableMachineConfig] = Field( + title="Teachable Machine Model Configs." + ) class SemanticSearchConfig(FrigateBaseModel):