From c34f72108542935e55b1271f5ac454e53b74bfd8 Mon Sep 17 00:00:00 2001 From: serengil Date: Fri, 4 Dec 2020 16:07:16 +0300 Subject: [PATCH] clean code for pre-built models --- README.md | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/README.md b/README.md index 01a6906..e4f6338 100644 --- a/README.md +++ b/README.md @@ -172,12 +172,10 @@ for model_name in models: DeepFace.verify("img1.jpg", "img2.jpg", model_name = model_name, model = model) #facial analysis -import json models = {} -models["emotion"] = DeepFace.build_model('Emotion') -models["age"] = DeepFace.build_model('Age') -models["gender"] = DeepFace.build_model('Gender') -models["race"] = DeepFace.build_model('Race') +actions = ['Age', 'Gender', 'Emotion', 'Race'] +for action in actions: + models[action.lower()] = DeepFace.build_model(action) DeepFace.analyze("img1.jpg", models=models) ```