github/deepface
1
1
Fork 0
mirror of https://github.com/serengil/deepface.git synced 2025-07-24 02:40:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

mtcnn is a must dependency

Browse Source
This commit is contained in:
Şefik Serangil 2020-09-10 07:59:52 +03:00
parent dc7ac1fe77
commit f3de735d72
3 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
deepface/commons/functions.py
View File

@ -18,6 +18,7 @@ import tensorflow as tf
import keras import keras
import bz2 import bz2
from deepface.commons import distance from deepface.commons import distance
from mtcnn import MTCNN
def loadBase64Img(uri): def loadBase64Img(uri):
encoded_data = uri.split(',')[1] encoded_data = uri.split(',')[1]
@ -275,9 +276,7 @@ def detect_face(img, detector_backend = 'opencv', grayscale = False, enforce_det
raise ValueError("Face could not be detected. Please confirm that the picture is a face photo or consider to set enforce_detection param to False.") raise ValueError("Face could not be detected. Please confirm that the picture is a face photo or consider to set enforce_detection param to False.")
elif detector_backend == 'mtcnn': elif detector_backend == 'mtcnn':
from mtcnn import MTCNN #this is not a must library within deepface
mtcnn_detector = MTCNN() mtcnn_detector = MTCNN()
detections = mtcnn_detector.detect_faces(img) detections = mtcnn_detector.detect_faces(img)
@ -432,7 +431,7 @@ def align_face(img, detector_backend = 'opencv'):
return img #return img anyway return img #return img anyway
elif detector_backend == 'mtcnn': elif detector_backend == 'mtcnn':
from mtcnn import MTCNN
mtcnn_detector = MTCNN() mtcnn_detector = MTCNN()
detections = mtcnn_detector.detect_faces(img) detections = mtcnn_detector.detect_faces(img)

1
requirements.txt
View File

@ -9,3 +9,4 @@ keras>=2.2.0
Flask>=1.1.2 Flask>=1.1.2
lightgbm>=2.3.1 lightgbm>=2.3.1
dlib>=19.20.0 dlib>=19.20.0
mtcnn>=0.1.0

2
setup.py
View File

@ -20,5 +20,5 @@ setuptools.setup(
"Operating System :: OS Independent", "Operating System :: OS Independent",
], ],
python_requires='>=3.5.5', python_requires='>=3.5.5',
install_requires=["numpy>=1.14.0", "pandas>=0.23.4", "tqdm>=4.30.0", "gdown>=3.10.1", "Pillow>=5.2.0", "opencv-python>=3.4.4", "tensorflow>=1.9.0", "keras>=2.2.0", "Flask>=1.1.2"] install_requires=["numpy>=1.14.0", "pandas>=0.23.4", "tqdm>=4.30.0", "gdown>=3.10.1", "Pillow>=5.2.0", "opencv-python>=3.4.4", "tensorflow>=1.9.0", "keras>=2.2.0", "Flask>=1.1.2", "mtcnn>=0.1.0"]
) )