mtcnn is a must

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

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)

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