diff --git a/deepface/commons/functions.py b/deepface/commons/functions.py index df535d7..135546d 100644 --- a/deepface/commons/functions.py +++ b/deepface/commons/functions.py @@ -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) diff --git a/requirements.txt b/requirements.txt index 65f5932..02f965e 100644 --- a/requirements.txt +++ b/requirements.txt @@ -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 \ No newline at end of file +Flask>=1.1.2 +mtcnn>=0.1.0 \ No newline at end of file