From 44078eddb87d27c453ecf5a2be4bcd38dbf495a9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=C5=9Eefik=20Serangil?= Date: Mon, 15 Jun 2020 16:11:57 +0300 Subject: [PATCH] ensemble pre-trained model --- deepface/DeepFace.py | 2 -- deepface/models/__init__.py | 0 {models => deepface/models}/face-recognition-ensemble-model.txt | 0 setup.py | 2 +- 4 files changed, 1 insertion(+), 3 deletions(-) create mode 100644 deepface/models/__init__.py rename {models => deepface/models}/face-recognition-ensemble-model.txt (100%) diff --git a/deepface/DeepFace.py b/deepface/DeepFace.py index 92a2d06..3eaa057 100644 --- a/deepface/DeepFace.py +++ b/deepface/DeepFace.py @@ -123,7 +123,6 @@ def verify(img1_path, img2_path='' #------------------------------- #find deepface path deepface_path = DeepFace.__file__ - #deepface_path = deepface_path.replace("\\", "/").replace("/deepface/DeepFace.py", "") deepface_path = deepface_path.replace("\\", "/").replace("/DeepFace.py", "") ensemble_model_path = deepface_path+"/models/face-recognition-ensemble-model.txt" @@ -677,7 +676,6 @@ def find(img_path, db_path #---------------------------------- #lightgbm model deepface_path = DeepFace.__file__ - #deepface_path = deepface_path.replace("\\", "/").replace("/deepface/DeepFace.py", "") deepface_path = deepface_path.replace("\\", "/").replace("/DeepFace.py", "") ensemble_model_path = deepface_path+"/models/face-recognition-ensemble-model.txt" deepface_ensemble = lgb.Booster(model_file = ensemble_model_path) diff --git a/deepface/models/__init__.py b/deepface/models/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/models/face-recognition-ensemble-model.txt b/deepface/models/face-recognition-ensemble-model.txt similarity index 100% rename from models/face-recognition-ensemble-model.txt rename to deepface/models/face-recognition-ensemble-model.txt diff --git a/setup.py b/setup.py index 78c0f14..ee0ee32 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.27", + version="0.0.28", author="Sefik Ilkin Serengil", author_email="serengil@gmail.com", description="Deep Face Analysis Framework for Face Recognition and Demography (Age, Gender, Emotion, Race)",