From 9aa19aa6fadbf25a35a9bb222c5c9de8d065399f Mon Sep 17 00:00:00 2001 From: vahidrezanezhad Date: Mon, 16 Feb 2026 11:45:56 +0100 Subject: [PATCH] cnn-rnn ocr inference: get input shape of model --- src/eynollah/eynollah_ocr.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/eynollah/eynollah_ocr.py b/src/eynollah/eynollah_ocr.py index 3c918e5..b5228ff 100644 --- a/src/eynollah/eynollah_ocr.py +++ b/src/eynollah/eynollah_ocr.py @@ -70,6 +70,7 @@ class Eynollah_ocr: self.model_zoo.get('ocr').to(self.device) else: self.model_zoo.load_model('ocr', '') + self.input_shape = self.model_zoo.get('ocr').input_shape[1:3] self.model_zoo.load_model('num_to_char') self.model_zoo.load_model('characters') self.end_character = len(self.model_zoo.get('characters', list)) + 2 @@ -823,8 +824,8 @@ class Eynollah_ocr: page_ns=page_ns, img_bin=img_bin, - image_width=512, - image_height=32, + image_width=self.input_shape[1], + image_height=self.input_shape[0], ) self.write_ocr(