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

Add model type

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

6
frigate/config/classification.py
View File

@ -34,10 +34,16 @@ class BirdClassificationConfig(FrigateBaseModel):
) )
class TeachableMachineModelType(str, Enum):
object = "object"
state = "state"
class TeachableMachineConfig(FrigateBaseModel): class TeachableMachineConfig(FrigateBaseModel):
enabled: bool = Field(default=True, title="Enable running the model.") enabled: bool = Field(default=True, title="Enable running the model.")
model_path: str = Field(title="Path to teachable machine tflite model.") model_path: str = Field(title="Path to teachable machine tflite model.")
labelmap_path: str = Field(title="Path to teachable machine labelmap.") labelmap_path: str = Field(title="Path to teachable machine labelmap.")
model_type: TeachableMachineModelType = Field(title="Type of model.")
class ClassificationConfig(FrigateBaseModel): class ClassificationConfig(FrigateBaseModel):