zishuo/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2025-09-26 19:41:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

YOLOv9 LPR model is not compatible (#20159)

Browse Source
This commit is contained in:
Nicolas Mowen
2025-09-21 17:51:45 -06:00
committed by GitHub
parent cd519ed1ad
commit 0e61d3f153
3 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
frigate/detectors/detection_runners.py
View File

@@ -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):

2
frigate/embeddings/onnx/lpr_embedding.py
View File

@@ -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):

1
frigate/embeddings/types.py
View File

@@ -12,3 +12,4 @@ class EnrichmentModelTypeEnum(str, Enum):
jina_v1 = "jina_v1"
jina_v2 = "jina_v2"
paddleocr = "paddleocr"
yolov9_license_plate = "yolov9_license_plate"