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

getting tasks dynamically

Browse Source
This commit is contained in:
Sefik Ilkin Serengil 2024-08-04 11:57:34 +01:00
parent 111c20cca5
commit 5c1bf67507
1 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
deepface/modules/modeling.py
View File

@ -45,11 +45,6 @@ def build_model(task: str, model_name: str) -> Any:
built model class built model class
""" """
tasks = ["facial_recognition", "spoofing", "facial_attribute", "face_detector"]
if task not in tasks:
raise ValueError(f"unimplemented task - {task}")
# singleton design pattern # singleton design pattern
global model_obj global model_obj
@ -89,8 +84,11 @@ def build_model(task: str, model_name: str) -> Any:
}, },
} }
if task not in models.keys():
raise ValueError(f"unimplemented task - {task}")
if not "model_obj" in globals(): if not "model_obj" in globals():
model_obj = {current_task: {} for current_task in tasks} model_obj = {current_task: {} for current_task in models.keys()}
if not model_name in model_obj[task].keys(): if not model_name in model_obj[task].keys():
model = models[task].get(model_name) model = models[task].get(model_name)