Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2025-09-23 17:52:05 +02:00
Code Issues 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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 3 additions and 1 deletions
Show Stats 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.paddleocr.value,
EnrichmentModelTypeEnum.jina_v1.value, EnrichmentModelTypeEnum.jina_v1.value,
EnrichmentModelTypeEnum.jina_v2.value, EnrichmentModelTypeEnum.jina_v2.value,
EnrichmentModelTypeEnum.yolov9_license_plate.value,
] ]
def __init__(self, session: ort.InferenceSession, cuda_device_id: int): 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( self.runner = get_optimized_runner(
os.path.join(self.download_path, self.model_file), os.path.join(self.download_path, self.model_file),
self.device, self.device,
model_type="yolov9", model_type=EnrichmentModelTypeEnum.yolov9_license_plate.value,
) )
def _preprocess_inputs(self, raw_inputs): 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_v1 = "jina_v1"
jina_v2 = "jina_v2" jina_v2 = "jina_v2"
paddleocr = "paddleocr" paddleocr = "paddleocr"
yolov9_license_plate = "yolov9_license_plate"