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

mtcnn is a must

Browse Source
This commit is contained in:
serengil 2020-09-29 15:16:47 +03:00
parent e6da083983
commit aa25d178c4
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
deepface/commons/functions.py
View File

@ -18,6 +18,7 @@ import tensorflow as tf
import keras
import bz2
from deepface.commons import distance
from mtcnn import MTCNN #0.1.0
def loadBase64Img(uri):
encoded_data = uri.split(',')[1]
@ -275,7 +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.")
elif detector_backend == 'mtcnn':
from mtcnn import MTCNN #0.1.0
mtcnn_detector = MTCNN()
detections = mtcnn_detector.detect_faces(img)
@ -431,7 +432,6 @@ def align_face(img, detector_backend = 'opencv'):
elif detector_backend == 'mtcnn':
from mtcnn import MTCNN #0.1.0
mtcnn_detector = MTCNN()
detections = mtcnn_detector.detect_faces(img)

3
requirements.txt
View File

@ -6,4 +6,5 @@ Pillow>=5.2.0
opencv-python>=3.4.4
tensorflow>=1.9.0
keras>=2.2.0
Flask>=1.1.2
Flask>=1.1.2
mtcnn>=0.1.0