Spaces:
Sleeping
Sleeping
mask arr
Browse files- identify_cards.py +3 -3
identify_cards.py
CHANGED
@@ -304,9 +304,9 @@ def get_not_white_mask(img):
|
|
304 |
l_channel = lab_patch[:, :, 0]
|
305 |
a_channel = lab_patch[:, :, 1]
|
306 |
b_channel = lab_patch[:, :, 2]
|
307 |
-
mask_l = cv2.threshold(l_channel,
|
308 |
-
mask_a = cv2.threshold(a_channel,
|
309 |
-
mask_b = cv2.threshold(b_channel,
|
310 |
mask_ab = cv2.bitwise_or(mask_a, mask_b)
|
311 |
text_mask = cv2.bitwise_and(mask_l, mask_ab)
|
312 |
return text_mask
|
|
|
304 |
l_channel = lab_patch[:, :, 0]
|
305 |
a_channel = lab_patch[:, :, 1]
|
306 |
b_channel = lab_patch[:, :, 2]
|
307 |
+
mask_l = cv2.threshold(l_channel, 170, 255, cv2.THRESH_BINARY_INV)[1]
|
308 |
+
mask_a = cv2.threshold(a_channel, 130, 255, cv2.THRESH_BINARY)[1]
|
309 |
+
mask_b = cv2.threshold(b_channel, 130, 255, cv2.THRESH_BINARY)[1]
|
310 |
mask_ab = cv2.bitwise_or(mask_a, mask_b)
|
311 |
text_mask = cv2.bitwise_and(mask_l, mask_ab)
|
312 |
return text_mask
|