Rename ONNXInference to FastPlateOCR

ankandrew · Apr 6, 2024 · a1a31dd · a1a31dd
1 parent 3ed7028
commit a1a31dd
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 1 deletion.
diff --git a/fast_plate_ocr/cli/__init__.py b/fast_plate_ocr/cli/__init__.py
diff --git a/fast_plate_ocr/inference/__init__.py b/fast_plate_ocr/inference/__init__.py
@@ -0,0 +1,7 @@
+"""
+FastPlateOCR package.
+"""
+
+from fast_plate_ocr.inference.onnx_inference import FastPlateOCR
+
+__all__ = ["FastPlateOCR"]
diff --git a/fast_plate_ocr/inference/onnx_inference.py b/fast_plate_ocr/inference/onnx_inference.py
@@ -41,7 +41,7 @@ def _load_image_from_source(source: str | list[str] | npt.NDArray) -> npt.NDArra
     raise ValueError("Unsupported input type. Only file path or numpy array is supported.")
 
 
-class ONNXInference:
+class FastPlateOCR:
     """
     ONNX inference class for performing license plates OCR.
     """