refactor: standardize public interface with __all__

Close issue #646

python/rapidocr/__init__.py

@@ -5,3 +5,15 @@ from .main import RapidOCR
 from .utils.load_image import LoadImageError
 from .utils.typings import EngineType, LangCls, LangDet, LangRec, ModelType, OCRVersion
 from .utils.vis_res import VisRes
+
+__all__ = [
+    "RapidOCR",
+    "LoadImageError",
+    "EngineType",
+    "LangCls",
+    "LangDet",
+    "LangRec",
+    "ModelType",
+    "OCRVersion",
+    "VisRes",
+]

python/rapidocr/cal_rec_boxes/__init__.py

@@ -2,3 +2,5 @@
 # @Author: SWHL
 # @Contact: liekkaskono@163.com
 from .main import CalRecBoxes
+
+__all__ = ["CalRecBoxes"]

python/rapidocr/ch_ppocr_cls/__init__.py

@@ -3,3 +3,5 @@
 # @Contact: liekkaskono@163.com
 from .main import TextClassifier
 from .utils import TextClsOutput
+
+__all__ = ["TextClassifier", "TextClsOutput"]

python/rapidocr/ch_ppocr_det/__init__.py

@@ -3,3 +3,5 @@
 # @Contact: liekkaskono@163.com
 from .main import TextDetector
 from .utils import TextDetOutput
+
+__all__ = ["TextDetector", "TextDetOutput"]

python/rapidocr/ch_ppocr_rec/__init__.py

@@ -3,3 +3,5 @@
 # @Contact: liekkaskono@163.com
 from .main import TextRecognizer
 from .typings import TextRecInput, TextRecOutput
+
+__all__ = ["TextRecognizer", "TextRecInput", "TextRecOutput"]

python/rapidocr/inference_engine/mnn/__init__.py

@@ -2,3 +2,5 @@
 # @Author: SWHL
 # @Contact: liekkaskono@163.com
 from .main import MNNError, MNNInferSession
+
+__all__ = ["MNNError", "MNNInferSession"]

python/rapidocr/inference_engine/onnxruntime/__init__.py

@@ -1,4 +1,6 @@
 # -*- encoding: utf-8 -*-
 # @Author: SWHL
 # @Contact: liekkaskono@163.com
-from .main import OrtInferSession
+from .main import ONNXRuntimeError, OrtInferSession
+
+__all__ = ["ONNXRuntimeError", "OrtInferSession"]

python/rapidocr/inference_engine/openvino/__init__.py

@@ -2,3 +2,5 @@
 # @Author: SWHL
 # @Contact: liekkaskono@163.com
 from .main import OpenVINOError, OpenVINOInferSession
+
+__all__ = ["OpenVINOError", "OpenVINOInferSession"]

python/rapidocr/inference_engine/paddle/__init__.py

@@ -2,3 +2,5 @@
 # @Author: SWHL
 # @Contact: liekkaskono@163.com
 from .main import PaddleInferError, PaddleInferSession
+
+__all__ = ["PaddleInferError", "PaddleInferSession"]

python/rapidocr/inference_engine/pytorch/__init__.py

@@ -2,3 +2,5 @@
 # @Author: SWHL
 # @Contact: liekkaskono@163.com
 from .main import TorchInferError, TorchInferSession
+
+__all__ = ["TorchInferError", "TorchInferSession"]

python/rapidocr/inference_engine/tensorrt/__init__.py

@@ -1,4 +1,6 @@
 # -*- encoding: utf-8 -*-
 # @Author: SWHL
 # @Contact: liekkaskono@163.com
-from .main import TRTInferSession
+from .main import TensorRTError, TRTInferSession
+
+__all__ = ["TensorRTError", "TRTInferSession"]