qurator-spk/sbb_binarization
1
0
Fork 0
mirror of https://github.com/qurator-spk/sbb_binarization.git synced 2025-06-09 04:10:00 +02:00
Code Issues Releases Wiki Activity

#42 is resolved

Browse source
This commit is contained in:
vahid 2022-08-25 23:51:16 +02:00
parent 2355d0ec8f
commit f11d0b0bf7
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
sbb_binarize/sbb_binarize.py
View file

@ -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]