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

Clenanup

Browse Source
This commit is contained in:
Nicolas Mowen 2025-09-18 14:13:10 -06:00
parent 62c1efc430
commit 157132973a
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
frigate/detectors/detection_runners.py
View File

@ -106,7 +106,7 @@ class CudaGraphRunner(BaseModelRunner):
# Import here to avoid circular imports # Import here to avoid circular imports
from frigate.detectors.detector_config import ModelTypeEnum from frigate.detectors.detector_config import ModelTypeEnum
from frigate.embeddings.types import EnrichmentModelTypeEnum from frigate.embeddings.types import EnrichmentModelTypeEnum
return model_type in [ return model_type in [
ModelTypeEnum.yolonas.value, ModelTypeEnum.yolonas.value,
EnrichmentModelTypeEnum.paddleocr.value, EnrichmentModelTypeEnum.paddleocr.value,
@ -173,7 +173,7 @@ class OpenVINOModelRunner(BaseModelRunner):
def is_complex_model(model_type: str) -> bool: def is_complex_model(model_type: str) -> bool:
# Import here to avoid circular imports # Import here to avoid circular imports
from frigate.embeddings.types import EnrichmentModelTypeEnum from frigate.embeddings.types import EnrichmentModelTypeEnum
return model_type in [EnrichmentModelTypeEnum.paddleocr.value] return model_type in [EnrichmentModelTypeEnum.paddleocr.value]
def __init__(self, model_path: str, device: str, model_type: str, **kwargs): def __init__(self, model_path: str, device: str, model_type: str, **kwargs):

3
frigate/embeddings/types.py
View File

@ -5,9 +5,10 @@ class EmbeddingTypeEnum(str, Enum):
thumbnail = "thumbnail" thumbnail = "thumbnail"
description = "description" description = "description"
class EnrichmentModelTypeEnum(str, Enum): class EnrichmentModelTypeEnum(str, Enum):
arcface = "arcface" arcface = "arcface"
facenet = "facenet" facenet = "facenet"
jina_v1 = "jina_v1" jina_v1 = "jina_v1"
jina_v2 = "jina_v2" jina_v2 = "jina_v2"
paddleocr = "paddleocr" paddleocr = "paddleocr"