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

Add enum for type of classification for objects

Browse Source
This commit is contained in:
Nicolas Mowen 2025-07-18 07:42:46 -06:00
parent 7ea288fe32
commit 81d4b26b01
2 changed files with 11 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

@ -3,6 +3,8 @@ from typing import Dict, List, Optional
from pydantic import ConfigDict, Field
from frigate.types import ObjectClassificationType
from .base import FrigateBaseModel
__all__ = [
@ -88,6 +90,10 @@ class CustomClassificationStateConfig(FrigateBaseModel):
class CustomClassificationObjectConfig(FrigateBaseModel):
objects: list[str] = Field(title="Object types to classify.")
classification_type: ObjectClassificationType = Field(
default=ObjectClassificationType.sub_label,
title="Type of classification that is applied.",
)
class CustomClassificationConfig(FrigateBaseModel):

5
frigate/types.py
View File

@ -25,6 +25,11 @@ class ModelStatusTypesEnum(str, Enum):
complete = "complete"
class ObjectClassificationType(str, Enum):
sub_label = "sub_label"
attribute = "attribute"
class TrackedObjectUpdateTypesEnum(str, Enum):
description = "description"
face = "face"