283375/arcaea-offline-ocr
1
0
Fork 0
mirror of https://github.com/283375/arcaea-offline-ocr.git synced 2025-04-18 13:00:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor: B30 ocr

Browse Source
This commit is contained in:
283375 2023-10-13 18:32:33 +08:00
parent 3400df2d52
commit 42bcd7b430
Signed by: 283375
SSH Key Fingerprint: SHA256:UcX0qg6ZOSDOeieKPGokA5h7soykG61nz2uxuQgVLSk
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -3,10 +3,14 @@ from typing import List, Optional, Tuple
import cv2 import cv2
import numpy as np import numpy as np
from PIL import Image
from ....crop import crop_xywh from ....crop import crop_xywh
from ....ocr import FixRects, ocr_digits_by_contour_knn, preprocess_hog from ....ocr import (
FixRects,
ocr_digits_by_contour_knn,
preprocess_hog,
resize_fill_square,
)
from ....phash_db import ImagePhashDatabase from ....phash_db import ImagePhashDatabase
from ....utils import construct_int_xywh_rect from ....utils import construct_int_xywh_rect
from ...shared import B30OcrResultItem from ...shared import B30OcrResultItem
@ -90,7 +94,7 @@ class ChieriBotV4Ocr:
jacket_roi = cv2.cvtColor( jacket_roi = cv2.cvtColor(
crop_xywh(component_bgr, jacket_rect), cv2.COLOR_BGR2GRAY crop_xywh(component_bgr, jacket_rect), cv2.COLOR_BGR2GRAY
) )
return self.phash_db.lookup_image(Image.fromarray(jacket_roi))[0] return self.phash_db.lookup_jacket(jacket_roi)[0]
def ocr_component_score_knn(self, component_bgr: cv2.Mat) -> int: def ocr_component_score_knn(self, component_bgr: cv2.Mat) -> int:
# sourcery skip: inline-immediately-returned-variable # sourcery skip: inline-immediately-returned-variable
@ -200,7 +204,7 @@ class ChieriBotV4Ocr:
digits = [] digits = []
for digit_rect in digit_rects: for digit_rect in digit_rects:
digit = crop_xywh(roi, digit_rect) digit = crop_xywh(roi, digit_rect)
digit = cv2.resize(digit, (20, 20)) digit = resize_fill_square(digit, 20)
digits.append(digit) digits.append(digit)
samples = preprocess_hog(digits) samples = preprocess_hog(digits)