YoursFunny/ArisuAutoSweeper
1
0
Fork 0
mirror of https://github.com/TheFunny/ArisuAutoSweeper synced 2026-10-10 07:02:06 +00:00
Code Releases Wiki Activity

feat: add white preprocessing option in OcrWhiteLetterOnComplexBackground

Browse Source
This commit is contained in:
YoursFunny
2026-07-06 22:40:26 +08:00
parent 622f2a467d
commit db1a470d94
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
module/ocr/ocr.py
+5 -2
View File
@@ -427,9 +427,12 @@ class Duration(Ocr):
class OcrWhiteLetterOnComplexBackground(Ocr): class OcrWhiteLetterOnComplexBackground(Ocr):
white_preprocess = True
def pre_process(self, image): def pre_process(self, image):
image = extract_white_letters(image, threshold=255) if self.white_preprocess:
image = cv2.merge([image, image, image]) image = extract_white_letters(image, threshold=255)
image = cv2.merge([image, image, image])
return image return image
def detect_and_ocr(self, *args, **kwargs): def detect_and_ocr(self, *args, **kwargs):