diff --git a/src/eynollah/eynollah_ocr.py b/src/eynollah/eynollah_ocr.py index 4d3ad0f..5e0712e 100644 --- a/src/eynollah/eynollah_ocr.py +++ b/src/eynollah/eynollah_ocr.py @@ -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') diff --git a/src/eynollah/ocrd_cli_recognize.py b/src/eynollah/ocrd_cli_recognize.py index 229d91d..2aa6d2f 100644 --- a/src/eynollah/ocrd_cli_recognize.py +++ b/src/eynollah/ocrd_cli_recognize.py @@ -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'],