283375/arcaea-offline-ocr
1
0
Fork 0
mirror of https://github.com/283375/arcaea-offline-ocr.git synced 2025-07-01 12:26:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor: module structure

Browse Source
This commit is contained in:
283375
2023-10-10 01:30:12 +08:00
parent d13076c667
commit 6a19ead8d1
3 changed files with 11 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/arcaea_offline_ocr/device/ocr.py
View File

@ -10,18 +10,19 @@ from ..ocr import (
preprocess_hog,
resize_fill_square,
)
from ..phash_db import ImagePHashDatabase
from .roi.extractor import DeviceRoiExtractor
from .roi.masker import DeviceRoiMasker
from ..phash_db import ImagePhashDatabase
from .common import DeviceOcrResult
from .rois.extractor import DeviceRoisExtractor
from .rois.masker import DeviceRoisMasker
class DeviceOcr:
def __init__(
self,
extractor: DeviceRoiExtractor,
masker: DeviceRoiMasker,
extractor: DeviceRoisExtractor,
masker: DeviceRoisMasker,
knn_model: cv2.ml.KNearest,
phash_db: ImagePHashDatabase,
phash_db: ImagePhashDatabase,
):
self.extractor = extractor
self.masker = masker