From 6036478889165ee4e0d0f5dd06fbf7cd72147e6f Mon Sep 17 00:00:00 2001 From: Konstantin Baierer Date: Mon, 1 Mar 2021 17:49:20 +0100 Subject: [PATCH] typo: s,enhancemnet,enhancement,g --- qurator/eynollah/eynollah.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/qurator/eynollah/eynollah.py b/qurator/eynollah/eynollah.py index 73c85d4..2b30f9f 100644 --- a/qurator/eynollah/eynollah.py +++ b/qurator/eynollah/eynollah.py @@ -119,7 +119,7 @@ class Eynollah: self.logger = getLogger('eynollah') self.dir_models = dir_models - self.model_dir_of_enhancemnet = dir_models + "/model_enhancement.h5" + self.model_dir_of_enhancement = dir_models + "/model_enhancement.h5" self.model_dir_of_col_classifier = dir_models + "/model_scale_classifier.h5" self.model_region_dir_p = dir_models + "/model_main_covid19_lr5-5_scale_1_1_great.h5" self.model_region_dir_p2 = dir_models + "/model_main_home_corona3_rot.h5" @@ -149,7 +149,7 @@ class Eynollah: def predict_enhancement(self, img): self.logger.debug("enter predict_enhancement") - model_enhancement, _ = self.start_new_session_and_model(self.model_dir_of_enhancemnet) + model_enhancement, _ = self.start_new_session_and_model(self.model_dir_of_enhancement) img_height_model = model_enhancement.layers[len(model_enhancement.layers) - 1].output_shape[1] img_width_model = model_enhancement.layers[len(model_enhancement.layers) - 1].output_shape[2]