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

Typo Mistake

Browse Source
This commit is contained in:
Uria Franko 2020-04-21 16:24:47 +03:00
parent 58b0b0a037
commit c6300b53e0
1 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
deepface/DeepFace.py
View File

@ -207,7 +207,7 @@ def analyze(img_path, actions = [], models = {}):
pbar = tqdm(range(0,len(actions)), desc='Finding actions') pbar = tqdm(range(0,len(actions)), desc='Finding actions')
action_idx = 0 action_idx = 0
img_244 = None # Set to prevent re-detection img_224 = None # Set to prevent re-detection
#for action in actions: #for action in actions:
for index in pbar: for index in pbar:
action = actions[index] action = actions[index]
@ -240,20 +240,20 @@ def analyze(img_path, actions = [], models = {}):
resp_obj += emotion_obj resp_obj += emotion_obj
elif action == 'age': elif action == 'age':
if img_244 is None: if img_224 is None:
img_244 = functions.detectFace(img_path, (224, 224), False) #just emotion model expects grayscale images img_224 = functions.detectFace(img_path, (224, 224), False) #just emotion model expects grayscale images
#print("age prediction") #print("age prediction")
age_predictions = age_model.predict(img_244)[0,:] age_predictions = age_model.predict(img_224)[0,:]
apparent_age = Age.findApparentAge(age_predictions) apparent_age = Age.findApparentAge(age_predictions)
resp_obj += "\"age\": %s" % (apparent_age) resp_obj += "\"age\": %s" % (apparent_age)
elif action == 'gender': elif action == 'gender':
if img_244 is None: if img_224 is None:
img_244 = functions.detectFace(img_path, (224, 224), False) #just emotion model expects grayscale images img_224 = functions.detectFace(img_path, (224, 224), False) #just emotion model expects grayscale images
#print("gender prediction") #print("gender prediction")
gender_prediction = gender_model.predict(img_244)[0,:] gender_prediction = gender_model.predict(img_224)[0,:]
if np.argmax(gender_prediction) == 0: if np.argmax(gender_prediction) == 0:
gender = "Woman" gender = "Woman"
@ -263,9 +263,9 @@ def analyze(img_path, actions = [], models = {}):
resp_obj += "\"gender\": \"%s\"" % (gender) resp_obj += "\"gender\": \"%s\"" % (gender)
elif action == 'race': elif action == 'race':
if img_244 is None: if img_224 is None:
img_244 = functions.detectFace(img_path, (224, 224), False) #just emotion model expects grayscale images img_224 = functions.detectFace(img_path, (224, 224), False) #just emotion model expects grayscale images
race_predictions = race_model.predict(img_244)[0,:] race_predictions = race_model.predict(img_224)[0,:]
race_labels = ['asian', 'indian', 'black', 'white', 'middle eastern', 'latino hispanic'] race_labels = ['asian', 'indian', 'black', 'white', 'middle eastern', 'latino hispanic']
sum_of_predictions = race_predictions.sum() sum_of_predictions = race_predictions.sum()