From c67b25e3da69e9ccdb65a735fb15b42c672562fd Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Fri, 23 May 2025 05:21:54 -0600 Subject: [PATCH] Add model type --- frigate/config/classification.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/frigate/config/classification.py b/frigate/config/classification.py index dea991d92..76e87a06a 100644 --- a/frigate/config/classification.py +++ b/frigate/config/classification.py @@ -34,10 +34,16 @@ class BirdClassificationConfig(FrigateBaseModel): ) +class TeachableMachineModelType(str, Enum): + object = "object" + state = "state" + + 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.") + model_type: TeachableMachineModelType = Field(title="Type of model.") class ClassificationConfig(FrigateBaseModel):