From 4fd31b1e9b13e0bc1697258fc0e48500f4d7aa8e Mon Sep 17 00:00:00 2001 From: 283375 Date: Thu, 12 Oct 2023 17:53:51 +0800 Subject: [PATCH] impr: module `__init__` entries --- src/arcaea_offline_ocr/device/__init__.py | 2 ++ src/arcaea_offline_ocr/device/rois/__init__.py | 3 +++ src/arcaea_offline_ocr/device/rois/definition/__init__.py | 1 + src/arcaea_offline_ocr/device/rois/extractor/common.py | 4 ++-- 4 files changed, 8 insertions(+), 2 deletions(-) create mode 100644 src/arcaea_offline_ocr/device/rois/__init__.py diff --git a/src/arcaea_offline_ocr/device/__init__.py b/src/arcaea_offline_ocr/device/__init__.py index e69de29..bd6cd44 100644 --- a/src/arcaea_offline_ocr/device/__init__.py +++ b/src/arcaea_offline_ocr/device/__init__.py @@ -0,0 +1,2 @@ +from .common import DeviceOcrResult +from .ocr import DeviceOcr diff --git a/src/arcaea_offline_ocr/device/rois/__init__.py b/src/arcaea_offline_ocr/device/rois/__init__.py new file mode 100644 index 0000000..e73f32a --- /dev/null +++ b/src/arcaea_offline_ocr/device/rois/__init__.py @@ -0,0 +1,3 @@ +from .definition import * +from .extractor import * +from .masker import * diff --git a/src/arcaea_offline_ocr/device/rois/definition/__init__.py b/src/arcaea_offline_ocr/device/rois/definition/__init__.py index 4083875..37f6ef1 100644 --- a/src/arcaea_offline_ocr/device/rois/definition/__init__.py +++ b/src/arcaea_offline_ocr/device/rois/definition/__init__.py @@ -1 +1,2 @@ from .auto import * +from .common import DeviceRois diff --git a/src/arcaea_offline_ocr/device/rois/extractor/common.py b/src/arcaea_offline_ocr/device/rois/extractor/common.py index e672e8b..a90a0a4 100644 --- a/src/arcaea_offline_ocr/device/rois/extractor/common.py +++ b/src/arcaea_offline_ocr/device/rois/extractor/common.py @@ -5,9 +5,9 @@ from ..definition.common import DeviceRois class DeviceRoisExtractor: - def __init__(self, img: cv2.Mat, sizes: DeviceRois): + def __init__(self, img: cv2.Mat, rois: DeviceRois): self.img = img - self.sizes = sizes + self.sizes = rois def __construct_int_rect(self, rect): return tuple(round(r) for r in rect)