administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-05-03 12:07:40 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Add basic config for defining a teachable machine model

Browse Source
This commit is contained in:
Nicolas Mowen 2025-05-23 05:18:22 -06:00
parent 5dd30b273a
commit 6c6be1e048
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
frigate/config/classification.py
View File

@ -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): class ClassificationConfig(FrigateBaseModel):
bird: BirdClassificationConfig = Field( bird: BirdClassificationConfig = Field(
default_factory=BirdClassificationConfig, title="Bird classification config." default_factory=BirdClassificationConfig, title="Bird classification config."
) )
teachable_machine: Dict[str, TeachableMachineConfig] = Field(
title="Teachable Machine Model Configs."
)
class SemanticSearchConfig(FrigateBaseModel): class SemanticSearchConfig(FrigateBaseModel):