283375/arcaea-offline-ocr
1
0
Fork 0
mirror of https://github.com/283375/arcaea-offline-ocr.git synced 2025-04-16 03:50:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore!: basic mask module fixes

Browse Source
This commit is contained in:
283375 2023-08-05 02:28:20 +08:00
parent fd10c68c49
commit 8feb955cde
1 changed files with 2 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/arcaea_offline_ocr/mask.py
View File

@ -47,7 +47,7 @@ BYD_MIN_HSV = array([170, 50, 50], uint8)
BYD_MAX_HSV = array([179, 210, 198], uint8)
def mask_gray_new(__img_bgr: Mat):
def mask_gray(__img_bgr: Mat):
# bgr_value_equal_mask = all(__img_bgr[:, 1:] == __img_bgr[:, :-1], axis=1)
bgr_value_equal_mask = max(__img_bgr, axis=2) - min(__img_bgr, axis=2) <= 5
img_bgr = __img_bgr.copy()
@ -57,17 +57,9 @@ def mask_gray_new(__img_bgr: Mat):
return cv2.inRange(img_bgr, GRAY_MIN_BGR, GRAY_MAX_BGR)
def mask_gray(img_hsv: Mat):
mask = cv2.inRange(img_hsv, GRAY_MIN_HSV, GRAY_MAX_HSV)
mask = cv2.dilate(mask, (2, 2))
return mask
def mask_white(img_hsv: Mat):
mask = cv2.inRange(img_hsv, WHITE_MIN_HSV, WHITE_MAX_HSV)
mask = cv2.dilate(
mask, (5, 5), borderType=cv2.BORDER_CONSTANT | cv2.BORDER_ISOLATED
)
mask = cv2.dilate(mask, cv2.getStructuringElement(cv2.MORPH_RECT, (2, 2)))
return mask