diff --git a/frigate/detectors/detection_runners.py b/frigate/detectors/detection_runners.py index 9babbdef3..8f3c826f2 100644 --- a/frigate/detectors/detection_runners.py +++ b/frigate/detectors/detection_runners.py @@ -112,6 +112,7 @@ class CudaGraphRunner(BaseModelRunner): EnrichmentModelTypeEnum.paddleocr.value, EnrichmentModelTypeEnum.jina_v1.value, EnrichmentModelTypeEnum.jina_v2.value, + EnrichmentModelTypeEnum.yolov9_license_plate.value, ] def __init__(self, session: ort.InferenceSession, cuda_device_id: int): diff --git a/frigate/embeddings/onnx/lpr_embedding.py b/frigate/embeddings/onnx/lpr_embedding.py index f6d230ca4..e485bf611 100644 --- a/frigate/embeddings/onnx/lpr_embedding.py +++ b/frigate/embeddings/onnx/lpr_embedding.py @@ -262,7 +262,7 @@ class LicensePlateDetector(BaseEmbedding): self.runner = get_optimized_runner( os.path.join(self.download_path, self.model_file), self.device, - model_type="yolov9", + model_type=EnrichmentModelTypeEnum.yolov9_license_plate.value, ) def _preprocess_inputs(self, raw_inputs): diff --git a/frigate/embeddings/types.py b/frigate/embeddings/types.py index 2ed404437..32cbe5dd0 100644 --- a/frigate/embeddings/types.py +++ b/frigate/embeddings/types.py @@ -12,3 +12,4 @@ class EnrichmentModelTypeEnum(str, Enum): jina_v1 = "jina_v1" jina_v2 = "jina_v2" paddleocr = "paddleocr" + yolov9_license_plate = "yolov9_license_plate"