mirror of
https://github.com/qurator-spk/eynollah.git
synced 2026-04-30 19:22:03 +02:00
.
This commit is contained in:
parent
d2aae35446
commit
7f8bfc9945
2 changed files with 3 additions and 0 deletions
|
|
@ -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.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:
|
||||
self.model_zoo.load_model('trocr_processor')
|
||||
|
|
|
|||
|
|
@ -30,6 +30,7 @@ class EynollahRecognizeProcessor(Processor):
|
|||
assert self.parameter
|
||||
model_zoo = EynollahModelZoo(basedir=self.parameter['models'])
|
||||
assert self.parameter
|
||||
print(self.parameter)
|
||||
self.eynollah_ocr = Eynollah_ocr(
|
||||
model_zoo=model_zoo,
|
||||
tr_ocr=self.parameter['tr_ocr'],
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue