From f11d0b0bf741253c55930c34e58e7e10718cb652 Mon Sep 17 00:00:00 2001 From: vahid Date: Thu, 25 Aug 2022 23:51:16 +0200 Subject: [PATCH] #42 is resolved --- sbb_binarize/sbb_binarize.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sbb_binarize/sbb_binarize.py b/sbb_binarize/sbb_binarize.py index 2f94785..8960354 100644 --- 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]