qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2025-10-06 14:39:55 +02:00
Code Issues Projects Releases Wiki Activity

default cnn-rnn and transformer ocr models have changed to model_eynollah_ocr_cnnrnn_20250904 and model_eynollah_ocr_trocr_20250919 respectively

Browse source
This commit is contained in:
vahidrezanezhad 2025-09-21 16:33:14 +02:00 • committed by kba
parent 6bbdfe1074
commit 554f3988c9
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/eynollah/eynollah.py
View file

@ -317,9 +317,9 @@ class Eynollah:
#"/eynollah-textline_20210425" #"/eynollah-textline_20210425"
self.model_textline_dir = dir_models + "/modelens_textline_0_1__2_4_16092024" self.model_textline_dir = dir_models + "/modelens_textline_0_1__2_4_16092024"
if self.ocr and self.tr: if self.ocr and self.tr:
self.model_ocr_dir = dir_models + "/trocr_model_ens_of_3_checkpoints_201124" self.model_ocr_dir = dir_models + "/model_eynollah_ocr_trocr_20250919"
elif self.ocr and not self.tr: elif self.ocr and not self.tr:
self.model_ocr_dir = dir_models + "/model_eynollah_ocr_cnnrnn_20250805" self.model_ocr_dir = dir_models + "/model_eynollah_ocr_cnnrnn_20250904"
if self.tables: if self.tables:
if self.light_version: if self.light_version:
self.model_table_dir = dir_models + "/modelens_table_0t4_201124" self.model_table_dir = dir_models + "/modelens_table_0t4_201124"
@ -5226,7 +5226,7 @@ class Eynollah_ocr:
if self.model_name: if self.model_name:
self.model_ocr_dir = self.model_name self.model_ocr_dir = self.model_name
else: else:
self.model_ocr_dir = dir_models + "/trocr_model_ens_of_3_checkpoints_201124" self.model_ocr_dir = dir_models + "/model_eynollah_ocr_trocr_20250919"
self.model_ocr = VisionEncoderDecoderModel.from_pretrained(self.model_ocr_dir) self.model_ocr = VisionEncoderDecoderModel.from_pretrained(self.model_ocr_dir)
self.model_ocr.to(self.device) self.model_ocr.to(self.device)
if not batch_size: if not batch_size:
@ -5238,7 +5238,7 @@ class Eynollah_ocr:
if self.model_name: if self.model_name:
self.model_ocr_dir = self.model_name self.model_ocr_dir = self.model_name
else: else:
self.model_ocr_dir = dir_models + "/model_eynollah_ocr_cnnrnn_20250805" self.model_ocr_dir = dir_models + "/model_eynollah_ocr_cnnrnn_20250904"
model_ocr = load_model(self.model_ocr_dir , compile=False) model_ocr = load_model(self.model_ocr_dir , compile=False)
self.prediction_model = tf.keras.models.Model( self.prediction_model = tf.keras.models.Model(