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)
|