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

fix: previous refactor issues

Browse Source
This commit is contained in:
283375
2023-08-13 16:01:31 +08:00
parent c640754d87
commit ea2b663e83
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/arcaea_offline_ocr/ocr.py
View File

@ -44,7 +44,7 @@ def preprocess_hog(digit_rois):
def ocr_digits_by_contour_samples(__roi_gray: Mat, size: Tuple[int, int]):
roi = __roi_gray.copy()
contours = cv2.findContours(roi, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_NONE)
contours, _ = cv2.findContours(roi, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_NONE)
rects = sorted([cv2.boundingRect(c) for c in contours], key=lambda r: r[0])
digit_rois = [cv2.resize(crop_xywh(roi, rect), size) for rect in rects]
return preprocess_hog(digit_rois)