sui_feng/AzurLaneAutoScript
1
0
Fork 0
mirror of https://gitee.com/sui-feng-cb/AzurLaneAutoScript1 synced 2026-03-09 18:39:04 +08:00
Code

Fix: pre_process of LevelOcr may return NoneType

Browse Source
This commit is contained in:
noname94 2020-10-20 16:58:21 +09:00
parent 1ec5d66be2
commit 90d1b09bce
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
module/combat/level.py
View File

@ -97,6 +97,7 @@ class LevelOcr(Digit):
# Ruturn an empty image if 'L' is not found.
letter_l = np.nonzero(image[2:15, :].max(axis=0) < 127)[0]
if len(letter_l):
return image[:, letter_l[0] + 17:]
else:
return np.array([[255]], dtype=np.uint8)
first_digit = letter_l[0] + 17
if first_digit + 3 < LV_GRID_MAIN.button_shape[0]:
return image[:, first_digit:]
return np.array([[255]], dtype=np.uint8)