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

continue to use paddleocr v3 text detection model for large

Browse Source 
v5 was not finding text on multi-line plates at all in testing
This commit is contained in:
Josh Hawkins 2025-09-17 14:43:08 -05:00
parent 26178444f3
commit e6ccd40dbb
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -32,7 +32,7 @@ class PaddleOCRDetection(BaseEmbedding):
device: str = "AUTO",
):
model_file = (
"detection_v5-large.onnx"
"detection_v3-large.onnx"
if model_size == "large"
else "detection_v5-small.onnx"
)
@ -40,7 +40,7 @@ class PaddleOCRDetection(BaseEmbedding):
model_name="paddleocr-onnx",
model_file=model_file,
download_urls={
model_file: f"https://github.com/hawkeye217/paddleocr-onnx/raw/refs/heads/master/models/v5/{model_file}"
model_file: f"https://github.com/hawkeye217/paddleocr-onnx/raw/refs/heads/master/models/{'v3' if model_size == 'large' else 'v5'}/{model_file}"
},
)
self.requestor = requestor