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

Add: OCR models for cnocr==1.2.2

Browse Source
This commit is contained in:
LmeSzinc
2020-07-14 23:59:48 +08:00
parent 41ac5c16b9
commit b55abd1178
10 changed files with 10492 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
module/base/utils.py
View File

@@ -318,20 +318,19 @@ def color_similarity_2d(image, color):
return image
def extract_letters(image, letter=(255, 255, 255), back=(0, 0, 0)):
def extract_letters(image, letter=(255, 255, 255), threshold=128):
"""Set letter color to black, set background color to white.
Args:
image: Shape (height, width, channel)
letter (tuple): Letter RGB.
back (tuple): Background RGB.
threshold (int):
Returns:
np.ndarray: Shape (height, width)
"""
image = color_similarity_2d(np.array(image), color=letter)
back = color_similarity(back, letter)
image = (255.0 - image) * (1 + back / 255)
image = (255.0 - image) * (255.0 / threshold)
image[image > 255] = 255
return image