saimemrekanat commited on
Commit
6750432
1 Parent(s): 3e63f01

face detect test

Browse files
Files changed (1) hide show
  1. vtoonify_model.py +2 -1
vtoonify_model.py CHANGED
@@ -52,8 +52,9 @@ class Model():
52
  }
53
 
54
  self.landmarkpredictor = self._create_dlib_landmark_model()
 
55
  self.parsingpredictor = self._create_parsing_model()
56
- self.pspencoder = self._load_encoder()
57
  self.transform = transforms.Compose([
58
  transforms.ToTensor(),
59
  transforms.Normalize(mean=[0.5, 0.5, 0.5],std=[0.5,0.5,0.5]),
 
52
  }
53
 
54
  self.landmarkpredictor = self._create_dlib_landmark_model()
55
+ self.cnn_model = self._create_dlib_landmark_model()
56
  self.parsingpredictor = self._create_parsing_model()
57
+ self.pspencoder = self._load_encoder()
58
  self.transform = transforms.Compose([
59
  transforms.ToTensor(),
60
  transforms.Normalize(mean=[0.5, 0.5, 0.5],std=[0.5,0.5,0.5]),