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

chore: code linting

Browse Source
This commit is contained in:
283375
2023-11-03 00:50:45 +08:00
parent 76e6adbbf7
commit 17f6c2bac7
7 changed files with 27 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/arcaea_offline_ocr/device/ocr.py
View File

@ -67,8 +67,9 @@ class DeviceOcr:
roi = self.masker.score(self.extractor.score)
contours, _ = cv2.findContours(roi, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_NONE)
for contour in contours:
x, y, w, h = cv2.boundingRect(contour)
if h < roi.shape[0] * 0.6:
if (
cv2.boundingRect(contour)[3] < roi.shape[0] * 0.6
): # h < score_component_h * 0.6
roi = cv2.fillPoly(roi, [contour], [0])
return ocr_digits_by_contour_knn(roi, self.knn_model)