qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2025-06-07 19:35:01 +02:00
Code Issues Projects Releases Wiki Activity

typo: s,enhancemnet,enhancement,g

Browse source
This commit is contained in:
Konstantin Baierer 2021-03-01 17:49:20 +01:00
parent 5f2867bfcf
commit 6036478889
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
qurator/eynollah/eynollah.py
View file

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