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

fix: FixRects.split_connected

Browse Source
This commit is contained in:
283375 2023-09-05 01:30:25 +08:00
parent 15ead84772
commit 7f62fd20b1
Signed by: 283375
SSH Key Fingerprint: SHA256:UcX0qg6ZOSDOeieKPGokA5h7soykG61nz2uxuQgVLSk
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/arcaea_offline_ocr/ocr.py
View File

@ -95,7 +95,7 @@ class FixRects:
for i in range(img_cropped.shape[1]): for i in range(img_cropped.shape[1]):
col = img_cropped[:, i] col = img_cropped[:, i]
white_pixels[rx + border_ignore + i] = np.count_nonzero(col > 200) white_pixels[rx + border_ignore + i] = np.count_nonzero(col > 200)
least_white_pixels = min(list(white_pixels.values())) least_white_pixels = min(v for v in white_pixels.values() if v > 0)
x_values = [ x_values = [
x x
for x, pixel in white_pixels.items() for x, pixel in white_pixels.items()
@ -113,7 +113,7 @@ class FixRects:
# split the rect # split the rect
new_rects.extend( new_rects.extend(
[(rx, ry, x_mid, rh), (rx + x_mid, ry, rw - x_mid, rh)] [(rx, ry, x_mid - rx, rh), (x_mid, ry, rx + rw - x_mid, rh)]
) )
return_rects = deepcopy(rects) return_rects = deepcopy(rects)
@ -175,7 +175,7 @@ def ocr_digit_samples_knn(__samples, knn_model: cv2_ml_KNearest, k: int = 4):
_, results, _, _ = knn_model.findNearest(__samples, k) _, results, _, _ = knn_model.findNearest(__samples, k)
result_list = [int(r) for r in results.ravel()] result_list = [int(r) for r in results.ravel()]
result_str = "".join(str(r) for r in result_list if r > -1) result_str = "".join(str(r) for r in result_list if r > -1)
return int(result_str) return int(result_str) if result_str else 0
def ocr_digits_by_contour_get_samples(__roi_gray: Mat, size: int): def ocr_digits_by_contour_get_samples(__roi_gray: Mat, size: int):