preprocess

main.py

@@ -25,6 +25,7 @@ from identify_cards import (
     SUIT_TEMPLATE_PATH,
     determine_and_correct_orientation,
     find_rank_candidates,
+    get_not_white_mask,
     get_suit_from_image_rules,
     load_suit_templates,
     save_img_with_rect,
@@ -213,14 +214,11 @@ async def analyze_image(image_paths: list[UploadFile]):
                 # img = Image.fromarray(
                 #     cv2.cvtColor(candidate["img"], cv2.COLOR_BGR2RGB)
                 # )
-                custom_config = (
-                    r"--psm 10 -c tessedit_char_whitelist=0123456789AKQJ"
-                )
-
+                img = candidate["img"]
+                text_mask = get_not_white_mask(img)
+                masked_img = cv2.bitwise_and(img, img, mask=text)
                 # result = reader.readtext(candidate["img"])
-                result = reader.readtext(
-                    candidate["img"], text_threshold=0.4
-                )
+                result = reader.readtext(masked_img)
                 print(result)
                 if len(result) > 0:
                     ocr_results.append(result)