Spaces:
Runtime error
Runtime error
Commit
·
815e61d
1
Parent(s):
cd4be53
face detect test
Browse files
vtoonify/model/encoder/align_all_parallel.py
CHANGED
@@ -33,7 +33,7 @@ def get_landmark(filepath, predictor):
|
|
33 |
:return: np.array shape=(68, 2)
|
34 |
"""
|
35 |
detector = dlib.get_frontal_face_detector()
|
36 |
-
cnn_face_detector = dlib.cnn_face_detection_model_v1(
|
37 |
|
38 |
if type(filepath) == str:
|
39 |
img = dlib.load_rgb_image(filepath)
|
|
|
33 |
:return: np.array shape=(68, 2)
|
34 |
"""
|
35 |
detector = dlib.get_frontal_face_detector()
|
36 |
+
cnn_face_detector = dlib.cnn_face_detection_model_v1('localmodel/mmod_human_face_detector.dat') # Load the MMod CNN model
|
37 |
|
38 |
if type(filepath) == str:
|
39 |
img = dlib.load_rgb_image(filepath)
|