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

Return Built Models

Browse Source
This commit is contained in:
Uria Franko 2020-04-16 11:32:56 +03:00
parent d9c7ad6b7a
commit 1fcbd36464
1 changed files with 25 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
deepface/DeepFace.py
View File

@ -23,7 +23,7 @@ from deepface.commons import functions, realtime, distance as dst
def analyze_init(models = []): def analyze_init(models = []):
#--------------------------------- #---------------------------------
built_models = {}
#if a specific target is not passed, then find them all #if a specific target is not passed, then find them all
if len(models) == 0: if len(models) == 0:
models = ['emotion', 'age', 'gender', 'race'] models = ['emotion', 'age', 'gender', 'race']
@ -33,16 +33,17 @@ def analyze_init(models = []):
#--------------------------------- #---------------------------------
if 'emotion' in models: if 'emotion' in models:
emotion_model = Emotion.loadModel() built_models['emotion'] = Emotion.loadModel()
if 'age' in models: if 'age' in models:
age_model = Age.loadModel() built_models['age'] = Age.loadModel()
if 'gender' in models: if 'gender' in models:
gender_model = Gender.loadModel() built_models['gender'] = Gender.loadModel()
if 'race' in models: if 'race' in models:
race_model = Race.loadModel() built_models['race'] = Race.loadModel()
return built_models
def verify_init(model_name = 'VGG-Face'): def verify_init(model_name = 'VGG-Face'):
@ -63,6 +64,7 @@ def verify_init(model_name = 'VGG-Face'):
model = FbDeepFace.loadModel() model = FbDeepFace.loadModel()
else: else:
raise ValueError("Invalid model_name passed - ", model_name) raise ValueError("Invalid model_name passed - ", model_name)
return model
def verify(img1_path, img2_path='' def verify(img1_path, img2_path=''
@ -193,7 +195,8 @@ def verify(img1_path, img2_path=''
return resp_obj return resp_obj
#return resp_objects #return resp_objects
def analyze(img_path, actions= []):
def analyze(img_path, actions= [], models= {}):
if type(img_path) == list: if type(img_path) == list:
img_paths = img_path.copy() img_paths = img_path.copy()
@ -213,15 +216,27 @@ def analyze(img_path, actions= []):
#--------------------------------- #---------------------------------
if 'emotion' in actions: if 'emotion' in actions:
if 'emotion' in models:
emotion_model = models['emotion']
else:
emotion_model = Emotion.loadModel() emotion_model = Emotion.loadModel()
if 'age' in actions: if 'age' in actions:
if 'age' in models:
age_model = models['age']
else:
age_model = Age.loadModel() age_model = Age.loadModel()
if 'gender' in actions: if 'gender' in actions:
if 'gender' in models:
gender_model = models['gender']
else:
gender_model = Gender.loadModel() gender_model = Gender.loadModel()
if 'race' in actions: if 'race' in actions:
if 'race' in models:
race_model = models['race']
else:
race_model = Race.loadModel() race_model = Race.loadModel()
#--------------------------------- #---------------------------------