Bugfix: Make character name OCR not freeze program if fails, also improved read

This commit is contained in:
Gleed
2022-06-23 17:01:45 -04:00
committed by GitHub
parent 9ed7869ddd
commit 689a93a3f7
2 changed files with 21 additions and 16 deletions
+1 -1
View File
@@ -120,7 +120,7 @@ shrine_check=500,50,350,100
character_select=1033, 44, 226, 554
character_online_status=1033, 19, 226, 25
; character_sub_roi is with respect to matched active character template
character_name_sub_roi=6, 22, 186, 18
character_name_sub_roi=2, 20, 194, 22
cube_area_roi=167,209,113,152
cube_btn_roi=160,368,125,57
xp_bar_text=369,630,554,34
+20 -15
View File
@@ -61,21 +61,26 @@ def save_char_template():
x, y, w, h = Config().ui_roi["character_name_sub_roi"]
x, y = x + match.region[0], y + match.region[1]
char_template = cut_roi(img, [x, y, w, h])
ocr_result = ocr.image_to_text(
images = char_template,
model = "hover-eng_inconsolata_inv_th_fast",
psm = 6,
scale = 1.2,
crop_pad = False,
erode = False,
invert = False,
threshold = 0,
digits_only = False,
fix_regexps = False,
check_known_errors = False,
correct_words = False,
)[0]
Logger.debug(f"Saved character template: {ocr_result.text.splitlines()[0]}")
msg=""
try:
ocr_result = ocr.image_to_text(
images = cut_roi(img, [x, y, w, h]),
model = "hover-eng_inconsolata_inv_th_fast",
psm = 6,
scale = 1.2,
crop_pad = False,
erode = False,
invert = False,
threshold = 0,
digits_only = False,
fix_regexps = False,
check_known_errors = False,
correct_words = False,
)[0]
msg += f": {ocr_result.text.splitlines()[0]}"
except:
pass
Logger.debug(f"Saved character template{msg}")
else:
Logger.error("save_char_template: Could not save character template")
return