Skip to content

Commit

Permalink
Revert "Analyze Pass Models"
Browse files Browse the repository at this point in the history
This reverts commit b124c7b.
  • Loading branch information
uriafranko committed Apr 17, 2020
1 parent b124c7b commit 378417f
Showing 1 changed file with 46 additions and 3 deletions.
49 changes: 46 additions & 3 deletions deepface/DeepFace.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,52 @@
from deepface.commons import functions, realtime, distance as dst


def analyze_init(models = []):
#---------------------------------
built_models = {}
#if a specific target is not passed, then find them all
if len(models) == 0:
models = ['emotion', 'age', 'gender', 'race']

print("Models to initialize: ", models)

#---------------------------------

if 'emotion' in models:
built_models['emotion'] = Emotion.loadModel()

if 'age' in models:
built_models['age'] = Age.loadModel()

if 'gender' in models:
built_models['gender'] = Gender.loadModel()

if 'race' in models:
built_models['race'] = Race.loadModel()
return built_models


def verify_init(model_name = 'VGG-Face'):
if model_name == 'VGG-Face':
print("Loading %s model" % model_name)
model = VGGFace.loadModel()

elif model_name == 'OpenFace':
print("Loading %s model" % model_name)
model = OpenFace.loadModel()

elif model_name == 'Facenet':
print("Loading %s model" % model_name)
model = Facenet.loadModel()

elif model_name == 'DeepFace':
print("Loading %s model" % model_name)
model = FbDeepFace.loadModel()
else:
raise ValueError("Invalid model_name passed - ", model_name)
return model


def verify(img1_path, img2_path=''
, model_name ='VGG-Face', distance_metric = 'cosine', model = None):

Expand Down Expand Up @@ -303,16 +349,13 @@ def analyze(img_path, actions= [], models= {}):
return resp_obj
#return resp_objects


def detectFace(img_path):
img = functions.detectFace(img_path)[0] #detectFace returns (1, 224, 224, 3)
return img[:, :, ::-1] #bgr to rgb


def stream(db_path, model_name ='VGG-Face', distance_metric = 'cosine', enable_face_analysis = True):
realtime.analysis(db_path, model_name, distance_metric, enable_face_analysis)


#---------------------------

functions.initializeFolder()
Expand Down

0 comments on commit 378417f

Please sign in to comment.