mask arr

identify_cards.py

@@ -5,10 +5,12 @@ import os
 import cv2
 import numpy as np
 from PIL import Image
-# from transformers import pipeline
 
 from utils import arrange_hand
 
+# from transformers import pipeline
+
+
 # # --- グローバル変数としてTrOCRパイプラインを初期化 ---
 # print("TrOCRのAIモデルを読み込んでいます...（初回は数分かかります）")
 # try:
@@ -302,7 +304,7 @@ def get_not_white_mask(img):
     l_channel = lab_patch[:, :, 0]
     a_channel = lab_patch[:, :, 1]
     b_channel = lab_patch[:, :, 2]
-    mask_l = cv2.threshold(l_channel, 170, 255, cv2.THRESH_BINARY_INV)[1]
+    mask_l = cv2.threshold(l_channel, 160, 255, cv2.THRESH_BINARY_INV)[1]
     mask_a = cv2.threshold(a_channel, 120, 255, cv2.THRESH_BINARY)[1]
     mask_b = cv2.threshold(b_channel, 120, 255, cv2.THRESH_BINARY)[1]
     mask_ab = cv2.bitwise_or(mask_a, mask_b)
@@ -509,7 +511,7 @@ def filter_thickness(
             # print("    -> スートと判断し除外")
             continue
         # else:
-            # print("    -> ランク候補として採用")
+        # print("    -> ランク候補として採用")
 
         if cropped_img.size > 0:
             candidate["thickness"] = max_val