mirror of
https://github.com/qurator-spk/eynollah.git
synced 2025-07-13 04:49:56 +02:00
solving issue by loading model by directory as input
This commit is contained in:
parent
89e58910aa
commit
38bf0d8740
1 changed files with 6 additions and 2 deletions
|
@ -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'
|
||||
|
@ -2927,7 +2928,10 @@ class Eynollah:
|
|||
|
||||
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
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue