diff --git a/frigate/detectors/detection_runners.py b/frigate/detectors/detection_runners.py index 332cf9488..f5b11cc56 100644 --- a/frigate/detectors/detection_runners.py +++ b/frigate/detectors/detection_runners.py @@ -106,7 +106,7 @@ class CudaGraphRunner(BaseModelRunner): # Import here to avoid circular imports from frigate.detectors.detector_config import ModelTypeEnum from frigate.embeddings.types import EnrichmentModelTypeEnum - + return model_type in [ ModelTypeEnum.yolonas.value, EnrichmentModelTypeEnum.paddleocr.value, @@ -173,7 +173,7 @@ class OpenVINOModelRunner(BaseModelRunner): def is_complex_model(model_type: str) -> bool: # Import here to avoid circular imports from frigate.embeddings.types import EnrichmentModelTypeEnum - + return model_type in [EnrichmentModelTypeEnum.paddleocr.value] def __init__(self, model_path: str, device: str, model_type: str, **kwargs): diff --git a/frigate/embeddings/types.py b/frigate/embeddings/types.py index 7fd7e43fa..2ed404437 100644 --- a/frigate/embeddings/types.py +++ b/frigate/embeddings/types.py @@ -5,9 +5,10 @@ class EmbeddingTypeEnum(str, Enum): thumbnail = "thumbnail" description = "description" + class EnrichmentModelTypeEnum(str, Enum): arcface = "arcface" facenet = "facenet" jina_v1 = "jina_v1" jina_v2 = "jina_v2" - paddleocr = "paddleocr" \ No newline at end of file + paddleocr = "paddleocr"