#42 is resolved

2 years ago · f11d0b0bf7
parent 2355d0ec8f
commit f11d0b0bf7
1 changed files with 1 additions and 1 deletions
--- a/sbb_binarize/sbb_binarize.py
+++ b/sbb_binarize/sbb_binarize.py
@ -54,7 +54,7 @@ class SbbBinarizer:
        del self.session

    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_width = model.layers[len(model.layers)-1].output_shape[2]
        n_classes = model.layers[len(model.layers)-1].output_shape[3]