issue 185

This commit is contained in:
Sefik Ilkin Serengil 2021-02-11 11:58:22 +03:00 committed by GitHub
parent bdebdc07f3
commit d37e8930fc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -116,6 +116,7 @@ def verify(img1_path, img2_path = '', model_name = 'VGG-Face', distance_metric =
else: else:
if model_name == 'Ensemble': if model_name == 'Ensemble':
Boosting.validate_model(model) Boosting.validate_model(model)
models = model.copy()
else: else:
models = {} models = {}
models[model_name] = model models[model_name] = model
@ -137,7 +138,7 @@ def verify(img1_path, img2_path = '', model_name = 'VGG-Face', distance_metric =
ensemble_features = [] ensemble_features = []
for i in model_names: for i in model_names:
custom_model = model[i] custom_model = models[i]
#decide input shape #decide input shape
input_shape = functions.find_input_shape(custom_model) input_shape = functions.find_input_shape(custom_model)