sui_feng/AzurLaneAutoScript
1
0
Fork 0
mirror of https://gitee.com/sui-feng-cb/AzurLaneAutoScript1 synced 2026-03-12 14:27:00 +08:00
Code Activity

Opt: Use existing function rgb2luma()

Browse Source
This commit is contained in:
LmeSzinc
2023-02-07 19:21:04 +08:00
parent 8d06ce4987
commit 6b4e62cb7e
1 changed files with 2 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
module/ocr/ocr.py
View File

@@ -107,9 +107,7 @@ class OcrYuv(Ocr):
@cached_property @cached_property
def letter_y(self): def letter_y(self):
arr = np.array([[self.letter]], dtype=np.uint8) arr = np.array([[self.letter]], dtype=np.uint8)
image = cv2.cvtColor(arr, cv2.COLOR_RGB2YUV) y = rgb2luma(arr)[0][0]
y, _, _ = cv2.split(image)
y = y[0][0]
return y return y
def pre_process(self, image): def pre_process(self, image):
@@ -120,8 +118,7 @@ class OcrYuv(Ocr):
Returns: Returns:
np.ndarray: Shape (width, height) np.ndarray: Shape (width, height)
""" """
image = cv2.cvtColor(image, cv2.COLOR_RGB2YUV) y = rgb2luma(image)
y, _, _ = cv2.split(image)
letter_y = (np.ones(y.shape) * self.letter_y).astype(np.uint8) letter_y = (np.ones(y.shape) * self.letter_y).astype(np.uint8)
diff = cv2.absdiff(y, letter_y) diff = cv2.absdiff(y, letter_y)
diff = cv2.multiply(diff, 255.0 / self.threshold) diff = cv2.multiply(diff, 255.0 / self.threshold)