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

impr: apply FixRects to default ocr method

Browse Source
This commit is contained in:
283375
2023-08-30 20:52:02 +08:00
parent c1140803b1
commit 90d268d8a6
2 changed files with 11 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,11 +1,11 @@
from math import floor
from typing import TYPE_CHECKING, List, Optional, Tuple
from typing import List, Optional, Tuple
import cv2
import numpy as np
from ....crop import crop_xywh
from ....ocr import ocr_digits_by_contour_knn, preprocess_hog
from ....ocr import FixRects, ocr_digits_by_contour_knn, preprocess_hog
from ....sift_db import SIFTDatabase
from ....types import Mat, cv2_ml_KNearest
from ....utils import construct_int_xywh_rect
@ -212,10 +212,12 @@ class ChieriBotV4Ocr:
digit_contours, _ = cv2.findContours(
roi, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE
)
digit_rects = sorted(
[cv2.boundingRect(c) for c in digit_contours],
key=lambda r: r[0],
digit_rects = [cv2.boundingRect(c) for c in digit_contours]
digit_rects = FixRects.connect_broken(
digit_rects, roi.shape[1], roi.shape[0]
)
digit_rects = FixRects.split_connected(roi, digit_rects)
digit_rects = sorted(digit_rects, key=lambda r: r[0])
digits = []
for digit_rect in digit_rects:
digit = crop_xywh(roi, digit_rect)