From a1a31dd4f46fafad57e06f5ba1a0f7b8189e8c0b Mon Sep 17 00:00:00 2001 From: ankandrew <61120139+ankandrew@users.noreply.github.com> Date: Sat, 6 Apr 2024 20:07:31 -0300 Subject: [PATCH] Rename ONNXInference to FastPlateOCR --- fast_plate_ocr/cli/__init__.py | 0 fast_plate_ocr/inference/__init__.py | 7 +++++++ fast_plate_ocr/inference/onnx_inference.py | 2 +- 3 files changed, 8 insertions(+), 1 deletion(-) create mode 100644 fast_plate_ocr/cli/__init__.py diff --git a/fast_plate_ocr/cli/__init__.py b/fast_plate_ocr/cli/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/fast_plate_ocr/inference/__init__.py b/fast_plate_ocr/inference/__init__.py index e69de29..6d42be9 100644 --- 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 index df6f38d..9e326c7 100644 --- 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. """