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

feat: ETERNAL rating class support

Browse Source
This commit is contained in:
283375 2024-03-20 15:53:10 +08:00
parent 17f6c2bac7
commit 00cd32dfdc
Signed by: 283375
SSH Key Fingerprint: SHA256:UcX0qg6ZOSDOeieKPGokA5h7soykG61nz2uxuQgVLSk
3 changed files with 19 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/arcaea_offline_ocr/device/ocr.py
View File

@ -80,6 +80,7 @@ class DeviceOcr:
self.masker.rating_class_prs(roi),
self.masker.rating_class_ftr(roi),
self.masker.rating_class_byd(roi),
self.masker.rating_class_etr(roi),
]
return max(enumerate(results), key=lambda i: np.count_nonzero(i[1]))[0]

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

@ -34,6 +34,9 @@ class DeviceRoisMaskerAutoT1(DeviceRoisMaskerAuto):
BYD_HSV_MIN = np.array([170, 50, 50], np.uint8)
BYD_HSV_MAX = np.array([179, 210, 198], np.uint8)
ETR_HSV_MIN = np.array([130, 60, 80], np.uint8)
ETR_HSV_MAX = np.array([140, 145, 180], np.uint8)
TRACK_LOST_HSV_MIN = np.array([170, 75, 90], np.uint8)
TRACK_LOST_HSV_MAX = np.array([175, 170, 160], np.uint8)
@ -87,6 +90,10 @@ class DeviceRoisMaskerAutoT1(DeviceRoisMaskerAuto):
def rating_class_byd(cls, roi_bgr: Mat) -> Mat:
return cls.mask_bgr_in_hsv(roi_bgr, cls.BYD_HSV_MIN, cls.BYD_HSV_MAX)
@classmethod
def rating_class_etr(cls, roi_bgr: Mat) -> Mat:
return cls.mask_bgr_in_hsv(roi_bgr, cls.ETR_HSV_MIN, cls.ETR_HSV_MAX)
@classmethod
def max_recall(cls, roi_bgr: Mat) -> Mat:
return cls.gray(roi_bgr)
@ -135,6 +142,9 @@ class DeviceRoisMaskerAutoT2(DeviceRoisMaskerAuto):
BYD_HSV_MIN = np.array([170, 50, 50], np.uint8)
BYD_HSV_MAX = np.array([179, 210, 198], np.uint8)
ETR_HSV_MIN = np.array([130, 60, 80], np.uint8)
ETR_HSV_MAX = np.array([140, 145, 180], np.uint8)
MAX_RECALL_HSV_MIN = np.array([125, 0, 0], np.uint8)
MAX_RECALL_HSV_MAX = np.array([145, 100, 150], np.uint8)
@ -186,6 +196,10 @@ class DeviceRoisMaskerAutoT2(DeviceRoisMaskerAuto):
def rating_class_byd(cls, roi_bgr: Mat) -> Mat:
return cls.mask_bgr_in_hsv(roi_bgr, cls.BYD_HSV_MIN, cls.BYD_HSV_MAX)
@classmethod
def rating_class_etr(cls, roi_bgr: Mat) -> Mat:
return cls.mask_bgr_in_hsv(roi_bgr, cls.ETR_HSV_MIN, cls.ETR_HSV_MAX)
@classmethod
def max_recall(cls, roi_bgr: Mat) -> Mat:
return cls.mask_bgr_in_hsv(

4
src/arcaea_offline_ocr/device/rois/masker/common.py
View File

@ -34,6 +34,10 @@ class DeviceRoisMasker:
def rating_class_byd(cls, roi_bgr: Mat) -> Mat:
raise NotImplementedError()
@classmethod
def rating_class_etr(cls, roi_bgr: Mat) -> Mat:
raise NotImplementedError()
@classmethod
def max_recall(cls, roi_bgr: Mat) -> Mat:
raise NotImplementedError()