283375/arcaea-offline-ocr
1
0
Fork 0
mirror of https://github.com/283375/arcaea-offline-ocr.git synced 2025-07-05 06:16:26 +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

6
src/arcaea_offline_ocr/b30/chieri/v4/ocr.py
View File

@ -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