qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2026-05-01 03:32:00 +02:00
Code Issues Projects Releases Wiki Activity

.

Browse source
This commit is contained in:
kba 2026-04-28 15:45:03 +02:00
parent d2aae35446
commit 7f8bfc9945
2 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/eynollah/eynollah_ocr.py
View file

@ -63,6 +63,8 @@ class Eynollah_ocr:
self.min_conf_value_of_textline_text = min_conf_value_of_textline_text if min_conf_value_of_textline_text else 0.3 self.min_conf_value_of_textline_text = min_conf_value_of_textline_text if min_conf_value_of_textline_text else 0.3
self.b_s = 2 if batch_size is None and tr_ocr else 8 if batch_size is None else batch_size self.b_s = 2 if batch_size is None and tr_ocr else 8 if batch_size is None else batch_size
print(batch_size)
print(self.b_s)
if tr_ocr: if tr_ocr:
self.model_zoo.load_model('trocr_processor') self.model_zoo.load_model('trocr_processor')

1
src/eynollah/ocrd_cli_recognize.py
View file

@ -30,6 +30,7 @@ class EynollahRecognizeProcessor(Processor):
assert self.parameter assert self.parameter
model_zoo = EynollahModelZoo(basedir=self.parameter['models']) model_zoo = EynollahModelZoo(basedir=self.parameter['models'])
assert self.parameter assert self.parameter
print(self.parameter)
self.eynollah_ocr = Eynollah_ocr( self.eynollah_ocr = Eynollah_ocr(
model_zoo=model_zoo, model_zoo=model_zoo,
tr_ocr=self.parameter['tr_ocr'], tr_ocr=self.parameter['tr_ocr'],