mirror of
https://github.com/283375/arcaea-offline-ocr.git
synced 2025-04-11 09:30:17 +00:00
refactor: module structure
This commit is contained in:
parent
d13076c667
commit
6a19ead8d1
@ -7,7 +7,7 @@ from PIL import Image
|
||||
|
||||
from ....crop import crop_xywh
|
||||
from ....ocr import FixRects, ocr_digits_by_contour_knn, preprocess_hog
|
||||
from ....phash_db import ImagePHashDatabase
|
||||
from ....phash_db import ImagePhashDatabase
|
||||
from ....types import Mat, cv2_ml_KNearest
|
||||
from ....utils import construct_int_xywh_rect
|
||||
from ...shared import B30OcrResultItem
|
||||
@ -20,7 +20,7 @@ class ChieriBotV4Ocr:
|
||||
self,
|
||||
score_knn: cv2_ml_KNearest,
|
||||
pfl_knn: cv2_ml_KNearest,
|
||||
phash_db: ImagePHashDatabase,
|
||||
phash_db: ImagePhashDatabase,
|
||||
factor: Optional[float] = 1.0,
|
||||
):
|
||||
self.__score_knn = score_knn
|
||||
@ -49,7 +49,7 @@ class ChieriBotV4Ocr:
|
||||
return self.__phash_db
|
||||
|
||||
@phash_db.setter
|
||||
def phash_db(self, phash_db: ImagePHashDatabase):
|
||||
def phash_db(self, phash_db: ImagePhashDatabase):
|
||||
self.__phash_db = phash_db
|
||||
|
||||
@property
|
||||
|
@ -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
|
||||
|
@ -35,7 +35,7 @@ def hamming_distance_sql_function(user_input, db_entry) -> int:
|
||||
)
|
||||
|
||||
|
||||
class ImagePHashDatabase:
|
||||
class ImagePhashDatabase:
|
||||
def __init__(self, db_path: str):
|
||||
with sqlite3.connect(db_path) as conn:
|
||||
self.hash_size = int(
|
||||
|
Loading…
x
Reference in New Issue
Block a user