Spaces:
Runtime error
Runtime error
Commit
·
6750432
1
Parent(s):
3e63f01
face detect test
Browse files- 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]),
|