solving issue by loading model by directory as input

pull/86/head
vahid 2 years ago
parent 89e58910aa
commit 38bf0d8740

@ -293,6 +293,7 @@ class Eynollah:
dir_out=self.dir_out, dir_out=self.dir_out,
image_filename=self.image_filename, image_filename=self.image_filename,
curved_line=self.curved_line, curved_line=self.curved_line,
textline_light = self.textline_light,
pcgts=self.pcgts) pcgts=self.pcgts)
def imread(self, grayscale=False, uint8=True): def imread(self, grayscale=False, uint8=True):
key = 'img' key = 'img'
@ -2927,7 +2928,10 @@ class Eynollah:
def our_load_model(self, model_file): def our_load_model(self, model_file):
try:
model = load_model(model_file, compile=False) model = load_model(model_file, compile=False)
except:
model = load_model(model_file , compile=False,custom_objects = {"PatchEncoder": PatchEncoder, "Patches": Patches})
return model return model

Loading…
Cancel
Save