From 3be387b92bc1897561a126a7627f8b486ac44ee6 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=C5=9Eefik=20Serangil?= Date: Thu, 17 Sep 2020 20:41:07 +0300 Subject: [PATCH] dlib is optional --- deepface/DeepFace.py | 3 ++- setup.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/deepface/DeepFace.py b/deepface/DeepFace.py index 115caea..8c51d07 100644 --- a/deepface/DeepFace.py +++ b/deepface/DeepFace.py @@ -18,7 +18,6 @@ import pickle from deepface import DeepFace from deepface.basemodels import VGGFace, OpenFace, Facenet, FbDeepFace, DeepID -from deepface.basemodels.DlibResNet import DlibResNet from deepface.extendedmodels import Age, Gender, Race, Emotion from deepface.commons import functions, realtime, distance as dst @@ -222,6 +221,7 @@ def verify(img1_path, img2_path = '', model_name ='VGG-Face', distance_metric = elif model_name == 'Dlib': print("Using Dlib ResNet model backend", distance_metric,"distance.") + from deepface.basemodels.DlibResNet import DlibResNet #this is not a must because it is very huge. model = DlibResNet() else: @@ -554,6 +554,7 @@ def find(img_path, db_path, model_name ='VGG-Face', distance_metric = 'cosine', model = DeepID.loadModel() elif model_name == 'Dlib': print("Using Dlib ResNet model backend", distance_metric,"distance.") + from deepface.basemodels.DlibResNet import DlibResNet #this is not a must because it is very huge model = DlibResNet() elif model_name == 'Ensemble': print("Ensemble learning enabled") diff --git a/setup.py b/setup.py index a6e4f07..5dec14a 100644 --- a/setup.py +++ b/setup.py @@ -5,7 +5,7 @@ with open("README.md", "r", encoding="utf-8") as fh: setuptools.setup( name="deepface", - version="0.0.37", + version="0.0.38", author="Sefik Ilkin Serengil", author_email="serengil@gmail.com", description="A Lightweight Face Recognition and Facial Attribute Analysis Framework (Age, Gender, Emotion, Race) for Python",