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

impr(Masker): expand score color range

Browse Source
This commit is contained in:
283375 2023-11-01 13:26:49 +08:00
parent bdf748bc63
commit 80404c436f
Signed by: 283375
SSH Key Fingerprint: SHA256:UcX0qg6ZOSDOeieKPGokA5h7soykG61nz2uxuQgVLSk
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
src/arcaea_offline_ocr/device/rois/masker/auto.py
View File

@ -13,8 +13,8 @@ class DeviceRoisMaskerAutoT1(DeviceRoisMaskerAuto):
GRAY_BGR_MIN = np.array([50] * 3, np.uint8) GRAY_BGR_MIN = np.array([50] * 3, np.uint8)
GRAY_BGR_MAX = np.array([160] * 3, np.uint8) GRAY_BGR_MAX = np.array([160] * 3, np.uint8)
WHITE_HSV_MIN = np.array([0, 0, 240], np.uint8) SCORE_HSV_MIN = np.array([0, 0, 180], np.uint8)
WHITE_HSV_MAX = np.array([179, 10, 255], np.uint8) SCORE_HSV_MAX = np.array([179, 255, 255], np.uint8)
PST_HSV_MIN = np.array([100, 50, 80], np.uint8) PST_HSV_MIN = np.array([100, 50, 80], np.uint8)
PST_HSV_MAX = np.array([100, 255, 255], np.uint8) PST_HSV_MAX = np.array([100, 255, 255], np.uint8)
@ -65,8 +65,8 @@ class DeviceRoisMaskerAutoT1(DeviceRoisMaskerAuto):
def score(cls, roi_bgr: Mat) -> Mat: def score(cls, roi_bgr: Mat) -> Mat:
return cv2.inRange( return cv2.inRange(
cv2.cvtColor(roi_bgr, cv2.COLOR_BGR2HSV), cv2.cvtColor(roi_bgr, cv2.COLOR_BGR2HSV),
cls.WHITE_HSV_MIN, cls.SCORE_HSV_MIN,
cls.WHITE_HSV_MAX, cls.SCORE_HSV_MAX,
) )
@classmethod @classmethod
@ -134,8 +134,8 @@ class DeviceRoisMaskerAutoT2(DeviceRoisMaskerAuto):
PFL_HSV_MIN = np.array([0, 0, 248], np.uint8) PFL_HSV_MIN = np.array([0, 0, 248], np.uint8)
PFL_HSV_MAX = np.array([179, 10, 255], np.uint8) PFL_HSV_MAX = np.array([179, 10, 255], np.uint8)
WHITE_HSV_MIN = np.array([0, 0, 240], np.uint8) SCORE_HSV_MIN = np.array([0, 0, 180], np.uint8)
WHITE_HSV_MAX = np.array([179, 10, 255], np.uint8) SCORE_HSV_MAX = np.array([179, 255, 255], np.uint8)
PST_HSV_MIN = np.array([100, 50, 80], np.uint8) PST_HSV_MIN = np.array([100, 50, 80], np.uint8)
PST_HSV_MAX = np.array([100, 255, 255], np.uint8) PST_HSV_MAX = np.array([100, 255, 255], np.uint8)
@ -186,8 +186,8 @@ class DeviceRoisMaskerAutoT2(DeviceRoisMaskerAuto):
def score(cls, roi_bgr: Mat) -> Mat: def score(cls, roi_bgr: Mat) -> Mat:
return cv2.inRange( return cv2.inRange(
cv2.cvtColor(roi_bgr, cv2.COLOR_BGR2HSV), cv2.cvtColor(roi_bgr, cv2.COLOR_BGR2HSV),
cls.WHITE_HSV_MIN, cls.SCORE_HSV_MIN,
cls.WHITE_HSV_MAX, cls.SCORE_HSV_MAX,
) )
@classmethod @classmethod