saimemrekanat commited on
Commit
8ecd0a6
1 Parent(s): f1a45cb

face detect test

Browse files
Files changed (1) hide show
  1. vtoonify_model.py +3 -3
vtoonify_model.py CHANGED
@@ -73,9 +73,9 @@ class Model():
73
 
74
  @staticmethod
75
  def _create_dlib_landmark_cnn_model():
76
- return dlib.shape_predictor(huggingface_hub.hf_hub_download(MODEL_REPO,
77
- 'models/mmod_human_face_detector.dat'))
78
-
79
  def _create_parsing_model(self):
80
  parsingpredictor = BiSeNet(n_classes=19)
81
  parsingpredictor.load_state_dict(torch.load(huggingface_hub.hf_hub_download(MODEL_REPO, 'models/faceparsing.pth'),
 
73
 
74
  @staticmethod
75
  def _create_dlib_landmark_cnn_model():
76
+ return dlib.cnn_face_detection_model_v1(huggingface_hub.hf_hub_download(MODEL_REPO,
77
+ 'models/mmod_human_face_detector.dat'))
78
+
79
  def _create_parsing_model(self):
80
  parsingpredictor = BiSeNet(n_classes=19)
81
  parsingpredictor.load_state_dict(torch.load(huggingface_hub.hf_hub_download(MODEL_REPO, 'models/faceparsing.pth'),