qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2025-10-27 07:44:12 +01:00
Code Issues Projects Releases Wiki Activity

solving issue by loading model by directory as input

Browse source
This commit is contained in:
vahid 2022-09-13 16:08:08 +02:00
parent 89e58910aa
commit 38bf0d8740
1 changed files with 6 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
qurator/eynollah/eynollah.py
View file

@ -293,6 +293,7 @@ class Eynollah:
dir_out=self.dir_out,
image_filename=self.image_filename,
curved_line=self.curved_line,
textline_light = self.textline_light,
pcgts=self.pcgts)
def imread(self, grayscale=False, uint8=True):
key = 'img'
@ -2926,8 +2927,11 @@ class Eynollah:
return polygons_of_images, img_revised_tab, text_regions_p_1_n, textline_mask_tot_d, regions_without_separators_d, regions_fully, regions_without_separators, polygons_of_marginals, contours_tables
def our_load_model(self, model_file):
model = load_model(model_file, compile=False)
try:
model = load_model(model_file, compile=False)
except:
model = load_model(model_file , compile=False,custom_objects = {"PatchEncoder": PatchEncoder, "Patches": Patches})
return model