saimemrekanat commited on
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(cnn_model_path) # Load the MMod CNN model
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)