From db1a470d942ede64a8557765902c3a48e95d4a18 Mon Sep 17 00:00:00 2001
From: YoursFunny <admin@yoursfunny.top>
Date: Mon, 6 Jul 2026 00:12:33 +0800
Subject: [PATCH] feat: add white preprocessing option in
 OcrWhiteLetterOnComplexBackground

---
 module/ocr/ocr.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/module/ocr/ocr.py b/module/ocr/ocr.py
index 5e95955..823ea4a 100644
--- a/module/ocr/ocr.py
+++ b/module/ocr/ocr.py
@@ -427,9 +427,12 @@ class Duration(Ocr):
 
 
 class OcrWhiteLetterOnComplexBackground(Ocr):
+    white_preprocess = True
+
     def pre_process(self, image):
-        image = extract_white_letters(image, threshold=255)
-        image = cv2.merge([image, image, image])
+        if self.white_preprocess:
+            image = extract_white_letters(image, threshold=255)
+            image = cv2.merge([image, image, image])
         return image
 
     def detect_and_ocr(self, *args, **kwargs):