Update default model to global-plates-mobile-vit-v2-model

ankandrew · Dec 8, 2024 · f8e8e71 · f8e8e71
1 parent d30ac4b
commit f8e8e71
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 6 deletions.
diff --git a/README.md b/README.md
@@ -56,7 +56,7 @@ from fast_alpr import ALPR
 # You can also initialize the ALPR with custom plate detection and OCR models.
 alpr = ALPR(
     detector_model="yolo-v9-t-384-license-plate-end2end",
-    ocr_model="european-plates-mobile-vit-v2-model",
+    ocr_model="global-plates-mobile-vit-v2-model",
 )
 
 # The "assets/test_image.png" can be found in repo root dit
@@ -78,7 +78,7 @@ from fast_alpr import ALPR
 # Initialize the ALPR
 alpr = ALPR(
     detector_model="yolo-v9-t-384-license-plate-end2end",
-    ocr_model="european-plates-mobile-vit-v2-model",
+    ocr_model="global-plates-mobile-vit-v2-model",
 )
 
 # Load the image

diff --git a/docs/quick_start.md b/docs/quick_start.md
@@ -10,7 +10,7 @@ from fast_alpr import ALPR
 # You can also initialize the ALPR with custom plate detection and OCR models.
 alpr = ALPR(
     detector_model="yolo-v9-t-384-license-plate-end2end",
-    ocr_model="european-plates-mobile-vit-v2-model",
+    ocr_model="global-plates-mobile-vit-v2-model",
 )
 
 # The "assets/test_image.png" can be found in repo root dit
@@ -38,7 +38,7 @@ from fast_alpr import ALPR
 # Initialize the ALPR
 alpr = ALPR(
     detector_model="yolo-v9-t-384-license-plate-end2end",
-    ocr_model="european-plates-mobile-vit-v2-model",
+    ocr_model="global-plates-mobile-vit-v2-model",
 )
 
 # Load the image

diff --git a/fast_alpr/alpr.py b/fast_alpr/alpr.py
@@ -47,7 +47,7 @@ def __init__(
         detector_conf_thresh: float = 0.4,
         detector_providers: Sequence[str | tuple[str, dict]] | None = None,
         detector_sess_options: ort.SessionOptions = None,
-        ocr_model: OcrModel | None = "european-plates-mobile-vit-v2-model",
+        ocr_model: OcrModel | None = "global-plates-mobile-vit-v2-model",
         ocr_device: Literal["cuda", "cpu", "auto"] = "auto",
         ocr_providers: Sequence[str | tuple[str, dict]] | None = None,
         ocr_sess_options: ort.SessionOptions | None = None,

diff --git a/test/test_alpr.py b/test/test_alpr.py
@@ -19,7 +19,7 @@ def test_default_alpr(img_path: Path, expected_plates: set[str]) -> None:
     im = cv2.imread(str(img_path))
     alpr = ALPR(
         detector_model="yolo-v9-t-384-license-plate-end2end",
-        ocr_model="european-plates-mobile-vit-v2-model",
+        ocr_model="global-plates-mobile-vit-v2-model",
     )
     actual_result = alpr.predict(im)
     actual_plates = {x.ocr.text for x in actual_result if x.ocr is not None}