github/deepface
1
1
Fork 0
mirror of https://github.com/serengil/deepface.git synced 2025-06-06 11:35:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

cleaner controls

Browse Source
This commit is contained in:
Sefik Ilkin Serengil 2024-08-04 12:02:24 +01:00
parent 5c1bf67507
commit 8e2caf6ede
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
deepface/modules/modeling.py
View File

@ -46,7 +46,7 @@ def build_model(task: str, model_name: str) -> Any:
"""
# singleton design pattern
global model_obj
global cached_models
models = {
"facial_recognition": {
@ -84,17 +84,17 @@ def build_model(task: str, model_name: str) -> Any:
},
}
if task not in models.keys():
if models.get(task) is None:
raise ValueError(f"unimplemented task - {task}")
if not "model_obj" in globals():
model_obj = {current_task: {} for current_task in models.keys()}
if not "cached_models" in globals():
cached_models = {current_task: {} for current_task in models.keys()}
if not model_name in model_obj[task].keys():
if cached_models[task].get(model_name) is None:
model = models[task].get(model_name)
if model:
model_obj[task][model_name] = model()
cached_models[task][model_name] = model()
else:
raise ValueError(f"Invalid model_name passed - {task}/{model_name}")
return model_obj[task][model_name]
return cached_models[task][model_name]