mirror of
https://github.com/blakeblackshear/frigate.git
synced 2025-07-30 13:48:07 +02:00
be8ee068e2
* Implement model training via ZMQ and add model states to represent training * Get model updates working * Improve toasts and model state * Clean up logging * Add back in
32 lines
829 B
Python
32 lines
829 B
Python
from enum import Enum
|
|
from typing import TypedDict
|
|
|
|
from frigate.camera import CameraMetrics
|
|
from frigate.data_processing.types import DataProcessorMetrics
|
|
from frigate.object_detection.base import ObjectDetectProcess
|
|
|
|
|
|
class StatsTrackingTypes(TypedDict):
|
|
camera_metrics: dict[str, CameraMetrics]
|
|
embeddings_metrics: DataProcessorMetrics | None
|
|
detectors: dict[str, ObjectDetectProcess]
|
|
started: int
|
|
latest_frigate_version: str
|
|
last_updated: int
|
|
processes: dict[str, int]
|
|
|
|
|
|
class ModelStatusTypesEnum(str, Enum):
|
|
not_downloaded = "not_downloaded"
|
|
downloading = "downloading"
|
|
downloaded = "downloaded"
|
|
error = "error"
|
|
training = "training"
|
|
complete = "complete"
|
|
|
|
|
|
class TrackedObjectUpdateTypesEnum(str, Enum):
|
|
description = "description"
|
|
face = "face"
|
|
lpr = "lpr"
|