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):