qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2025-11-09 22:24:13 +01:00
Code Issues Projects Releases Wiki Activity

missed changing override mechanism in eynollah_ocr

Browse source
This commit is contained in:
kba 2025-10-29 20:47:13 +01:00
parent 29c273685f
commit 4772fd17e2
1 changed files with 9 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

13
src/eynollah/eynollah_ocr.py
View file

@ -1,4 +1,9 @@
# FIXME: fix all of those...
# pyright: reportPossiblyUnboundVariable=false # pyright: reportPossiblyUnboundVariable=false
# pyright: reportOptionalMemberAccess=false
# pyright: reportArgumentType=false
# pyright: reportCallIssue=false
# pyright: reportOptionalSubscript=false
from logging import Logger, getLogger from logging import Logger, getLogger
from typing import Optional from typing import Optional
@ -53,7 +58,6 @@ class Eynollah_ocr:
self, self,
*, *,
model_zoo: EynollahModelZoo, model_zoo: EynollahModelZoo,
dir_xmls=None,
tr_ocr=False, tr_ocr=False,
batch_size: Optional[int]=None, batch_size: Optional[int]=None,
export_textline_images_and_text: bool=False, export_textline_images_and_text: bool=False,
@ -82,12 +86,13 @@ class Eynollah_ocr:
if tr_ocr: if tr_ocr:
self.model_zoo.load_model('trocr_processor') self.model_zoo.load_model('trocr_processor')
self.model_zoo.load_model('ocr', 'tr', model_path_override=model_name) self.model_zoo.load_model('ocr', 'tr')
self.model_zoo.get('ocr').to(self.device) self.model_zoo.get('ocr').to(self.device)
else: else:
self.model_zoo.load_model('ocr', '', model_path_override=model_name) self.model_zoo.load_model('ocr', '')
self.model_zoo.load_model('num_to_char') self.model_zoo.load_model('num_to_char')
self.end_character = len(self.model_zoo.load_model('characters')) + 2 self.model_zoo.load_model('characters')
self.end_character = len(self.model_zoo.get('characters', list)) + 2
@property @property
def device(self): def device(self):