mirror of
https://github.com/qurator-spk/sbb_binarization.git
synced 2025-06-07 19:35:04 +02:00
#42 is resolved
This commit is contained in:
parent
2355d0ec8f
commit
f11d0b0bf7
1 changed files with 1 additions and 1 deletions
|
@ -54,7 +54,7 @@ class SbbBinarizer:
|
||||||
del self.session
|
del self.session
|
||||||
|
|
||||||
def load_model(self, model_name):
|
def load_model(self, model_name):
|
||||||
model = load_model(join(self.model_dir, model_name), compile=False)
|
model = load_model(model_name, compile=False)
|
||||||
model_height = model.layers[len(model.layers)-1].output_shape[1]
|
model_height = model.layers[len(model.layers)-1].output_shape[1]
|
||||||
model_width = model.layers[len(model.layers)-1].output_shape[2]
|
model_width = model.layers[len(model.layers)-1].output_shape[2]
|
||||||
n_classes = model.layers[len(model.layers)-1].output_shape[3]
|
n_classes = model.layers[len(model.layers)-1].output_shape[3]
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue