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

Simplify model initialization

Browse Source
This commit is contained in:
Andrea Lanfranchi 2024-02-21 15:18:51 +01:00
parent adf6328cd9
commit 810dfac4fe
No known key found for this signature in database
GPG Key ID: 2D623A60BB279915
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
deepface/modules/modeling.py
View File

@ -43,8 +43,7 @@ def build_model(model_name: str) -> Any:
if not model_name in model_obj.keys(): if not model_name in model_obj.keys():
model = models.get(model_name) model = models.get(model_name)
if model: if model:
model = model() model_obj[model_name] = model()
model_obj[model_name] = model
else: else:
raise ValueError(f"Invalid model_name passed - {model_name}") raise ValueError(f"Invalid model_name passed - {model_name}")