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

dlib is optional

Browse Source
This commit is contained in:
Şefik Serangil 2020-09-17 20:41:07 +03:00
parent 33fe352930
commit 3be387b92b
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
deepface/DeepFace.py
View File

@ -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")

2
setup.py
View File

@ -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",