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

cnn-rnn ocr inference: get input shape of model

Browse source
This commit is contained in:
vahidrezanezhad 2026-02-16 11:45:56 +01:00
parent 68dd5eab62
commit 9aa19aa6fa
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/eynollah/eynollah_ocr.py
View file

@ -70,6 +70,7 @@ class Eynollah_ocr:
self.model_zoo.get('ocr').to(self.device) self.model_zoo.get('ocr').to(self.device)
else: else:
self.model_zoo.load_model('ocr', '') 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('num_to_char')
self.model_zoo.load_model('characters') self.model_zoo.load_model('characters')
self.end_character = len(self.model_zoo.get('characters', list)) + 2 self.end_character = len(self.model_zoo.get('characters', list)) + 2
@ -823,8 +824,8 @@ class Eynollah_ocr:
page_ns=page_ns, page_ns=page_ns,
img_bin=img_bin, img_bin=img_bin,
image_width=512, image_width=self.input_shape[1],
image_height=32, image_height=self.input_shape[0],
) )
self.write_ocr( self.write_ocr(